python - 数据框中特定行的总和( Pandas )

Question

我得到了一组以下数据:

week  A      B      C      D      E
1     243    857    393    621    194
2     644    576    534    792    207
3     946    252    453    547    436
4     560    100    864    663    949
5     712    734    308    385    303

我被要求找到指定行数/指定周数的每一列的总和，然后将这些数字绘制到条形图上以比较 A-E。

假设我有我需要的行(例如 df.iloc[2:4,:])，接下来我应该做什么？我的假设是我需要创建一个包含单行的掩码，其中包含每列的总和，但我不确定我该怎么做。

我知道如何执行最后一步(即 .plot(kind='bar')，我只需要知道中间步骤是什么以获得我需要的总和。

Answer 1

您可以使用位置 iloc、sum 和 Series.plot.bar 进行选择:

df.iloc[2:4].sum().plot.bar()

或者如果想按索引名称(此处为周)进行选择，请使用 loc :

df.loc[2:4].sum().plot.bar()

区别是 iloc 排除最后位置:

print (df.loc[2:4])
        A    B    C    D    E
week                         
2     644  576  534  792  207
3     946  252  453  547  436
4     560  100  864  663  949

print (df.iloc[2:4])
        A    B    C    D    E
week                         
3     946  252  453  547  436
4     560  100  864  663  949

---

如果需要还可以按位置过滤列:

df.iloc[2:4, :4].sum().plot.bar()  

按名称(周):

df.loc[2:4, list('ABCD')].sum().plot.bar()