按类别划分的直方图颜色

Question

我有一个带有 2 列“高度”和“类”的熊猫数据框，类是具有 3 个值 1,2 和 5 的列。 现在我想按类制作高度数据和颜色的直方图。 plot19_s["vegetation height"].plot.hist(bins = 10)

这是我的直方图

但现在我想通过直方图中颜色的变化来查看不同的类。

Answer 1

由于我不确定潜在的重复是否真的回答了这里的问题，这是一种使用 numpy.histogram 和 matplotlib bar 绘图生成堆叠直方图的方法。

import pandas as pd
import numpy as np;np.random.seed(1)
import matplotlib.pyplot as plt

df = pd.DataFrame({"x" : np.random.exponential(size=100),
                   "class" : np.random.choice([1,2,5],100)})

_, edges = np.histogram(df["x"], bins=10)
histdata = []; labels=[]
for n, group in df.groupby("class"):
    histdata.append(np.histogram(group["x"], bins=edges)[0])
    labels.append(n)

hist = np.array(histdata) 
histcum = np.cumsum(hist,axis=0)

plt.bar(edges[:-1],hist[0,:], width=np.diff(edges)[0],
            label=labels[0], align="edge")

for i in range(1,len(hist)):
    plt.bar(edges[:-1],hist[i,:], width=np.diff(edges)[0],
            bottom=histcum[i-1,:],label=labels[i], align="edge")

plt.legend(title="class")
plt.show()