python - 在具有增量名称的数据框列上应用公式

Question

我有一个数据框，其中包含一些以递增方式命名的列（例如 A_1、A_2、A_3、...）。我想对它们执行操作，例如：

A_1*1 + A_2*2 + A_3*3 + ...

有没有一种快速的方法来做到这一点，而不仅仅是写下所有列的名称（它们总共有 15 个）？

标签： pythonpandasdataframe

Answer 1

您还可以使用apply例如：

df['total'] = df.apply(lambda x: sum([x[col]*int(col.split('_')[-1]) for col in df.columns]),axis=1)

注意：因为我在df中添加了一列，所以不能两次执行相同的操作。