迭代並添加列熊貓

Question

我正在處理一個練習數據集，並被告知用其他人的總和創建一個新列，如下所示：

df["new column"] = df["column4"] + df["column5"] + df["column6"] + df["column7"] + etc....

我覺得這是低效的，因為它們甚至排成一行（第 4-9 列），我需要加快我的編碼速度（數據科學競賽即將到來！）。

我怎么能以更短/更有效的方式做到這一點！

我試過使用df['Total'] = sum(df.columns[4:9])類的東西的變體，但它不起作用

免責聲明：我是熊貓的新手

Answer 1

假設這些列彼此相鄰，請使用列切片：

df['new column'] = df.loc[:, 'column4':'column9'].sum(axis=1)

axis=1意味着你是跨列的總和，所以你正在計算每行的總數。

---

如果它們是分開的，您可以通過列表獲得它們的總和：

cols = ['column4', 'column5', 'column6', 'column7', 'column8', 'column9']
df['new column'] = df[cols].sum(axis=1)