使用现有列的总和创建列并循环

Question

我正在尝试使用数据框中的现有列生成新列。我的数据框有60列。需要基于数据帧中的数据生成60个新列。这些列应仅包含总和。 例如

Index column_a column_b month 
0.         1.     3.     Jan 
1.         2.     4.     Feb
2.         3.     4      Match

Index column_a column_b month  sum_a sum_b
0.         1       2.    Jan.   6.      10
1.         2.      4.    Feb.   6.      10
2.         3.      4     Match. 6.      10

请记住，我需要对此进行60列的迭代。将不胜感激。谢谢

Answer 1

首先创建您必须在其上计算总和的列的列表，然后在这些列上进行迭代

list_of_cols =['col_a','col_b']
for i in list_of_cols:
    name = 'sum_'+ i
    df[name] = df[i].sum()

我希望它能解决您的问题