python - 每两列的总和并在 Pandas 数据框中留下一列

Question

我的任务是这样的:

df=pd.DataFrame([(1,2,3,4,5,6),(1,2,3,4,5,6),(1,2,3,4,5,6)],columns=['a','b','c','d','e','f'])
Out:
    a b c d e f
0   1 2 3 4 5 6
1   1 2 3 4 5 6 
2   1 2 3 4 5 6

我想要做的是输出数据框看起来像这样:

Out
        s1 b s2  d  s3  f
    0   3  2  7  4  11  6
    1   3  2  7  4  11  6
    2   3  2  7  4  11  6

也就是说，分别对(a,b)、(c,d)、(e,f)列求和，保留最后一列并将结果列名称重命名为(s1,s2,s3)。谁能帮助解决 Pandas 中的这个问题？非常感谢。

Answer 1

您可以通过 iloc 按位置选择列，对每个 2 值求和，最后通过 f-strings 重命名列

i = 2
for x in range(0, len(df.columns), i):
    df.iloc[:, x] = df.iloc[:, x:x+i].sum(axis=1)
    df = df.rename(columns={df.columns[x]:f's{x // i + 1}'})
print (df)
   s1  b  s2  d  s3  f
0   3  2   7  4  11  6
1   3  2   7  4  11  6
2   3  2   7  4  11  6