我有一個熊貓數據框如下。我想從邏輯上檢查多列(A,B,C)并使用 OR 操作更新 D 列，并通過 E 列中的逗號分隔更新具有 true 的列名。我不確定如何實現它。任何線索都會有所幫助。A B C D ETrue True True True A,B,CTrue False True True A,CFalse False False False NAFalse False True True C 編輯 ：cols = ['A','B','C']df['D'] = np.where(df[cols].eq(True).any(1), True, False)我能夠想出 D 列，但不確定如何處理 E 列
2 回答
慕容森
TA貢獻1853條經驗 獲得超18個贊
使用DataFrame.dot+Series.str.rstrip和Series.replace：
cols = pd.Index(['A', 'B', 'C'])
df['E'] = df[cols].dot(cols + ',').str.rstrip(',').replace('', np.nan)
結果：
# print(df)
A B C D E
0 True True True True A,B,C
1 True False True True A,C
2 False False False False NaN
3 False False True True C
紅顏莎娜
TA貢獻1842條經驗 獲得超13個贊
另一種方法是使用stack
s = df[cols].stack()
df['F'] = s[s.eq(True)].reset_index(1).groupby(level=0)['level_1'].agg(','.join)
print(df)
A B C D E F
0 True True True True A,B,C A,B,C
1 True False True True A,C A,C
2 False False False False NaN NaN
3 False False True True C C
添加回答
舉報
0/150
提交
取消