Python堆叠直方图分组数据

Question

如何创建这样的堆叠直方图:

import numpy as np
import pylab as P
import matplotlib.pyplot as plt
import matplotlib
matplotlib.style.use('ggplot')

%pylab inline
pylab.rcParams['figure.figsize'] = (15, 4)
mu, sigma = 200, 25
x = mu + sigma*P.randn(1000,2)
n, bins, patches = P.hist(x, 10, normed=1, histtype='bar', stacked=True)
P.show()

对于分组数据。

import pandas as pd
df1 = pd.DataFrame({'A': [1,2,3,4,5,6,7,8,9],'group': [1,0,0,0,0,1,1,1,1]})

作为df1.A.hist(by=df1.group, bins='doane', stacked=True) 或

df1.A.hist(by=df1.group, bins='doane')

只会导致:

Answer 1

您可以 pivot您的 DataFrame，以便每个组位于不同的列中，然后生成直方图。

df1 = pd.DataFrame({'A': [1,2,3,4,5,6,7,8,9],'group': [1,0,0,0,0,1,1,1,1]})
df1.pivot(columns='group', values='A').plot.hist(stacked=True)

编辑:将 stacked=True 参数添加到 hist()