python - 在 Pandas 中创建其他列的总和值的计算列

Question

我有一个包含大约 60 列和以下结构的数据框:

    A    B  C           Y
0   12   1  0           1
1   13   1  0   [....]  0
2   14   0  1           1
3   15   1  0           0    
4   16   0  1           1

我想创建第 z 列，它将是 B 列到 Y 列的值的总和。

我该如何继续？

Answer 1

要在包含新列的同时创建数据框的副本，请使用assign

df.assign(Z=df.loc[:, 'B':'Y'].sum(1))

    A  B  C  Y  Z
0  12  1  0  1  2
1  13  1  0  0  1
2  14  0  1  1  2
3  15  1  0  0  1
4  16  0  1  1  2

---

要将它分配给同一个数据帧，就地使用

df['Z'] = df.loc[:, 'B':'Y'].sum(1)
df

    A  B  C  Y  Z
0  12  1  0  1  2
1  13  1  0  0  1
2  14  0  1  1  2
3  15  1  0  0  1
4  16  0  1  1  2