如何在python中总结N列？

Question

我有一只熊猫df,我想总结一下N列.df可能如下所示:

A B C D ... X

1 4 2 6     3
2 3 1 2     2 
3 1 1 2     4
4 2 3 5 ... 1

我想得到这样的df:

A Z

1 15
2 8
3 8
4 11

A变量不是索引,而是变量.

Answer 1

使用join新Series的创建sum无所有列A:

df = df[['A']].join(df.drop('A', 1).sum(axis=1).rename('Z'))

或提取柱A首先pop:

df = df.pop('A').to_frame().join(df.sum(axis=1).rename('Z'))

如果想按位置选择列,请使用iloc:

df = df.iloc[:, [0]].join(df.iloc[:, 1:].sum(axis=1).rename('Z'))

print (df)
   A   Z
0  1  15
1  2   8
2  3   8
3  4  11