绘制堆积百分比条形图 matplotlib

Question

假设我有以下包含两列的数据框： 标签：可以是 -1、0 或 1。 years_of_expereicen: 可以是 0,1,2,3,4,5,6,7,8,9

label   SSP_years_of_experience
22640   -1.0    5.0
181487  1.0 3.0
327672  0.0 9.0
254919  0.0 6.0
136942  1.0 10.0

我的目标是使用此数据框创建一个百分比堆积条形图，其中 x 轴是经验年数，条形图是不同的颜色，每种颜色都包含一年的经验值。换句话说，我们在 x 轴上有 10 个可能的值，然后为每个不同颜色的值映射到每个标签的三个条形图。 y 轴应以百分比为单位。

我知道如何在 R（使用 ggplot）中执行此操作，但我是 matplotlib 新手，对 python 有点陌生。

我可以将两列作为变量传递的奖励点（例如x，y）。更多关于如何在图表中将每个条形中的观察次数显示为文本的奖励积分。

Answer 1

如果你的数据框是pandas，试试：

exp_name = 'year_of_experience'
label_name = 'label'
new_df = (df.groupby(exp_name)[label_name]
            .value_counts(normalize=True)
            .sort_index()
            .unstack()
         )

new_df.plot.bar(stacked=True)

玩具数据框：

np.random.seed(0)
df = pd.DataFrame({'label': np.random.choice([-1,0,1], size=1000, replace=True),
                   'year_of_experience': np.random.randint(0,10, 1000)})

输出：