數據幀中特定行的總和（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

我被要求找到指定行/指定周數的每列的總和，然后將這些數字繪制到條形圖上以比較AE。

假設我有我需要的行（例如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()

Answer 2

您需要做的就是在您的數據子集上調用.sum() ：

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

返回：

week       7
A       1506
B        352
C       1317
D       1210
E       1385
dtype: int64

此外，對於繪圖，我認為你可以擺脫week列（因為周數的總和不太可能意味着什么）：

df.iloc[2:4,1:].sum().plot(kind='bar')
# or
df[list('ABCDE')].iloc[2:4].sum().plot(kind='bar')