使用以相同名称开头的列对 pandas 数据框行求和

Question

我有这个数据框：

import pandas as pd
d = {'1_col': [1, 2], '2_col': [3, 4], 'var1': [5,6]}
df = pd.DataFrame(data=d)
print(df)

看起来像这样：

    1_col  2_col  var1
0     1     3     5
1     2     4     6

我需要对名称以 _col 结尾的所有列求和，以便生成的 DataFrame 看起来像这样（名为 sum 的列是 1_col 列的总和和 2_col:

    1_col  2_col  var1 sum
0     1     3     5     4
1     2     4     6     6

pandas 中是否有一种方法可以对名称以“_col”结尾的所有列求和，而不是手动进行？

Answer 1

使用df.filter with df.sum:

In [1209]: df['sum'] = df.filter(like='_col').sum(1)

In [1210]: df
Out[1210]: 
   1_col  2_col  var1  sum
0      1      3     5    4
1      2      4     6    6