如何绘制熊猫数据框中列中位数的堆积条形图?

问题描述

所以我是一个学习 Pandas (python) 数据可视化的新手,我的任务是为 30 到 35 岁的年龄组创建一个周数和 CodeRevHrs 中位数的堆叠图。

以下是我在年龄列上应用过滤器提取数据的代码,下面是数据集的前五行

age_filter= agework [(agework["age"]>= 30 )&(agework["age"]<=35)] 
median_weekhrs= age_filter["Weekhrs"].median()
median_coderev= age_filter["CodeRevHrs"].median()

age_filter.head()

    CodeRevHrs  Weekhrs age
5   3.0          8.0    31.0
11  2.0         40.0    34.0
12  2.0         40.0    32.0
18  15.0        42.0    34.0
22  2.0         40.0    33.0

如何绘制带有中位数的堆积条形图?

请帮忙

解决方法

首先,过滤年龄(并将年龄转换为 int,因为它可以使标签更清晰):

df = agework.query('30 <= age <= 35')
df['age'] = df['age'].astype(int)

然后,您可以绘制每个年龄组中两个数量的中位数的条形图:

df.groupby('age').median().plot.bar(stacked=True)
plt.title('Median hours,by age')

顺便说一句,您可以对值的堆叠方式施加任意顺序。例如,如果您希望在底部使用 'Weekhrs',您可以说:

order = ['Weekhrs','CodeRevHrs']
df.groupby('age')[order].median().plot.bar(stacked=True)
plt.title('Median hours,by age')

现在,如果您想绘制整个过滤年龄范围内这些数量的总体中位数(如您所说:每个数量的一个数字),那么一种方式(众多)将是:

label = f"{df['age'].min()}-{df['age'].max()}"
df.median().drop('age').to_frame(label).T.plot.bar(stacked=True)
plt.title(f'Median hours for age {label}')