python - 在Python中优雅地求和命名的DataFrame列

Question

我正在尝试编写一个函数，对 pandas DataFrame 中的一组指定列进行求和。

首先，一些背景知识。每个数据都有一个列，其中包含名称(例如“var”)以及该名称旁边按顺序排列的数字(例如“var1、var2”)。我知道我可以使用以下代码对 5 列进行求和:

import pandas as pd
data = pd.read_csv('data_file.csv')
data['var_total'] = data.var1 + data.var2 + data.var3 + data.var4 + data.var5

但是，当您要对 var1-var30 求和时，这可能会重复。我认为必须有一些优雅的解决方案来更快地对它们求和，因为列名称是可预测的且统一的。有没有我可以编写的函数或内置的 pandas 函数可以让我更快地求和？

Answer 1

你可以这样做:

data['var_total'] = data.filter(regex='var[0-9]+').sum(axis=1)

这将首先过滤您的数据框，仅保留以 var 开头且后跟一个或多个数字的列。然后它将对过滤后的 DataFrame 进行求和。