我有這個資料框:
import pandas as pd
d = {'1_col': [1, 2], '2_col': [3, 4], 'var1': [5,6]}
df = pd.DataFrame(data=d)
print(df)
看起來像這樣:
1_col 2_col var1
0 1 3 5
1 2 4 6
我需要對名稱以 結尾的所有列求和_col,以便生成的 DataFrame 看起來像這樣(稱為的列sum是列的總和,1_col并且2_col:
1_col 2_col var1 sum
0 1 3 5 4
1 2 4 6 6
pandas 有沒有辦法對名稱以“_col”結尾的所有列求和,而不是手動進行?
uj5u.com熱心網友回復:
df.filter與 一起使用df.sum:
In [1209]: df['sum'] = df.filter(like='_col').sum(1)
In [1210]: df
Out[1210]:
1_col 2_col var1 sum
0 1 3 5 4
1 2 4 6 6
轉載請註明出處,本文鏈接:https://www.uj5u.com/ruanti/424780.html
