我需要一些幫助來弄清楚如何在同一數(shù)據(jù)框中將一列的值添加到其他列。例如，我有 df 隨機生成的數(shù)據(jù)：df = pd.DataFrame({'user_id': [123, 123, 246, 246, 135, 135], 'period': [1, 2, 1, 2, 1, 2], 'value1': [0, 5, 3, 1, 6, 3], 'value2': [5, 5, 2, 5, 4, 7], 'value3': [10, 15, 35, 25, 60, 50]}) | user_id |period| value1 | value2 | value3 | |---------|------|--------|--------|--------|0 | 123 | 1 | 0 | 5 | 10 |1 | 123 | 2 | 5 | 5 | 15 |2 | 246 | 1 | 3 | 2 | 35 |3 | 246 | 2 | 1 | 5 | 25 |4 | 135 | 1 | 6 | 4 | 60 |5 | 135 | 2 | 3 | 7 | 50 |所以在上面的 df 中，我將從“value3”列中獲取值，并將其添加到“value1”和“value2”中的值，同時保留其他列。df2 | user_id |period| value1 | value2 | value3 | |---------|------|--------|--------|--------|0 | 123 | 1 | 10 | 15 | 10 |1 | 123 | 2 | 20 | 20 | 15 |2 | 246 | 1 | 38 | 37 | 35 |3 | 246 | 2 | 26 | 30 | 25 |4 | 135 | 1 | 66 | 64 | 60 |5 | 135 | 2 | 53 | 57 | 50 |我嘗試對實際添加進行硬編碼，但遇到了錯誤代碼。我知道我可能遺漏了一些簡單的東西。有什么建議嗎？
2 回答
慕萊塢森
TA貢獻1810條經(jīng)驗 獲得超4個贊
指定axis=0何時將 Series 添加到 DataFrame。
cols = ['value1', 'value2']
df[cols] = df[cols].add(df['value3'], axis=0)
user_id period value1 value2 value3
0 123 1 10 15 10
1 123 2 20 20 15
2 246 1 38 37 35
3 246 2 26 30 25
4 135 1 66 64 60
5 135 2 53 57 50
GCT1015
TA貢獻1827條經(jīng)驗 獲得超4個贊
我們可以做numpy傳播
df.update(df[['value1','value2']]+df.value3.values[:,None])
df
user_id period value1 value2 value3
0 123 1 10 15 10
1 123 2 20 20 15
2 246 1 38 37 35
3 246 2 26 30 25
4 135 1 66 64 60
5 135 2 53 57 50
添加回答
舉報
0/150
提交
取消