我有一個像這樣有 4 列的數據框Asset1 Asset2 Asset3 Asset4 a b c d e f g h 我想創建一個列，itertools.combinations用于為我提供獨特組合的結果，因此理想情況下輸出將是： Asset1 Asset2 Asset3 Asset4 test a b c d [abc, abd, bcd, acd] e f g h [efg, efh, egh, fgh]我嘗試使用.join()和組合但不起作用。有任何想法嗎？
1 回答
慕森卡
TA貢獻1806條經驗 獲得超8個贊
歡迎來到 SO！
我建議使用lambdarow-wise ( axis=1)：
from itertools import combinations
import pandas as pd
df = pd.DataFrame({'Asset1':('a','e'), 'Asset2': ('b','f'), 'Asset3': ('c', 'g'), 'Asset4': ('d', 'h')})
df['combinations'] = df.apply(lambda r: list(combinations(r, 3)), axis=1)
print(df)
輸出：
Asset1 ... combinations
0 a ... [(a, b, c), (a, b, d), (a, c, d), (b, c, d)]
1 e ... [(e, f, g), (e, f, h), (e, g, h), (f, g, h)]
[2 rows x 5 columns]
list(combinations...如果您稍后只迭代組合，您也可以跳過- 這樣您將節省一些內存并將計算延遲到訪問的時刻df['combinations']：
df['combinations'] = df.apply(lambda r: combinations(r, 3), axis=1)
print(df)
然后你會在combinations列中得到一個非常神秘的對象：
Asset1 ... combinations
0 a ... <itertools.combinations object at 0x0000022392...
1 e ... <itertools.combinations object at 0x0000022392...
[2 rows x 5 columns]
添加回答
舉報
0/150
提交
取消