【发布时间】:2021-08-30 12:16:57
【问题描述】:
我有以下数据集:
data = [{'Title': 'Size', 'B': 20, 'C':30, 'D':15, 'E':15},
{'Title': 'Length', 'B': 50,'C':60, 'D':14,'E':15},
{'Title': 'Area', 'B': 4,'C':3, 'D':13,'E':15}]
df = pd.DataFrame(data)
我想对列“C”和“E”求和,并将包含求和值的列命名为“Total”。
最终结果应该是这样的:
我读到避免循环是个好主意,应该使用“列表理解”,所以我尝试了下面的代码,但它不起作用:
df2 = []
result = [(row[0]) for row in df[iloc.:,[2,4]].sum(axis=1).to_numpy()]
df2.append(result)
【问题讨论】:
标签: dataframe sum list-comprehension multiple-columns python-3.7