我試圖從某個列中減去選定的列，但它給了我一個錯誤“ValueError：期望一個一維陣列，得到一個形狀為 (3, 4) 的陣列”。我的 csv 檔案如下所示：
Number Nimi Summa Kino
0 1 Jan 30.5 12.5
1 2 Pets 30.5 12.5
2 3 Anni 30.5 12.5
因為用戶可以創建無限數量的列，所以我需要選擇除“Number, Nimi and Summa”之外的每一列。結果應該是需要從“Summa”中減去“Summa”之后的每一列。我希望這是有道理的。
df = pd.read_csv ('eelarve.csv')
columns_dont_want = ["Number", "Nimi", "Summa"]
select = [x for x in df.columns if x not in columns_dont_want]
df["Summa j?rel"] = df["Summa"] - df.loc[:, select]
df.to_csv('eelarve.csv', index=False)
uj5u.com熱心網友回復：
我在 df 中添加了更多包含資料的列，以確保下面的代碼有效。
import pandas as pd
df = pd.DataFrame({'Number': [1, 2, 3],
'Nimi': ['Jan', 'Pets', 'Anni'],
'Summa': [30.5, 30.5, 30.5],
'Kino': [12.5, 12.5, 12.5],
'Test1': [0, 1, 2],
'Test2': [0.1, 0.2, 0.3]
})
columns_dont_want = ["Number", "Nimi", "Summa"]
select = [x for x in df.columns if x not in columns_dont_want]
df['Summa j?rel'] = df['Summa'] - df[select].sum(axis=1)
輸出是：
Number Nimi Summa Kino Test1 Test2 Summa j?rel
0 1 Jan 30.5 12.5 0 0.1 17.9
1 2 Pets 30.5 12.5 1 0.2 16.8
2 3 Anni 30.5 12.5 2 0.3 15.7
uj5u.com熱心網友回復：
你可以做drop rsub：
df.drop(columns=columns_dont_want).rsub(df['Summa'], axis=0)
輸出：
Kino
0 18.0
1 18.0
2 18.0
要加入其他列：
out = (df[columns_dont_want]
.join(df.drop(columns=columns_dont_want)
.rsub(df['Summa'], axis=0))
)
輸出：
Number Nimi Summa Kino
0 1 Jan 30.5 18.0
1 2 Pets 30.5 18.0
2 3 Anni 30.5 18.0
轉載請註明出處，本文鏈接：https://www.uj5u.com/qianduan/455909.html