在Python中绘制从数字派生的比例数据

Question

我有一个数据框df，其中包含学生和非学生的年龄，如下所示：

Subject    Student    Age
001        yes        21
002        yes        45
003        no         61
004        no         37
...

我想绘制40岁以下每个组的比例。我可以使用plot(factor(age < 40) ~ Student, data = df)在R中做到这一点：

是否有一种方法可以在Python中复制，最好是使用matplotlib或seaborn？

Answer 1

没有内置选项可创建此类图。当然，您可以通过matplotlib来计算它的数量。

import numpy as np
import pandas as pd
import matplotlib.pyplot as plt

a = np.random.poisson(lam=40, size=6000)
b = ((a>50).astype(int)+np.random.rand(6000))>0.9
df = pd.DataFrame({"Subject" : np.arange(6000),
                   "Age" : a, "Student" : b})

df["Age>40"] = df["Age"] > 40

def propplot(x, y, data):
    xdata = data[[x,y]].groupby(x)
    xcount = xdata.count()

    fig, axes = plt.subplots(ncols=len(xcount), 
                             gridspec_kw={"width_ratios":list(xcount[y].values)})
    for ax, (n,grp) in zip(axes, xdata):

        ycount = grp.groupby(y).count().T
        ycount /= float(ycount.values.sum())
        ycount.plot.bar(stacked=True, ax=ax, width=1, legend=False)
        ax.set_xlabel(n)
        ax.set_xlim(-.5,.5)
        ax.set_ylim(0,1)
        ax.set_xticks([])
    axes[0].set_ylabel(y)
    axes[0].legend(ncol=100, title=y, loc=(0,1.02))
    fig.text(0.5,0.02, x)

propplot("Student", "Age>40", df)
plt.show()