思路:
根据columns获取columns列表然后运用apply和匿名函数进行求和,最后再获取一次列表将合计列一起筛选出来
import pandas as pd
data = pd.read_excel("大西瓜.xlsx")
data1 = data.copy()
data1["业绩合计"] = data1[data1.columns[data1.columns.str.contains("分月业绩")]].apply(lambda x: x.sum(), axis=1)
#根据提取含“分月业绩”字符串的行
a = data1[data1.columns[data1.columns.str.contains("分月业绩")]]
#形成列表
a1 = (data1.columns[data1.columns.str.contains("分月业绩")]).tolist()
a1.insert(0,"业绩合计")
#形成新的df
A = data1[a1]
数据源&结果展示:
数据源:
结果: