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

Question

我有一个包含学生和非学生年龄的数据框df，看起来像这样：

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

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

有没有办法在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()