如何根據(jù)python中DataFrame的列值選擇列名？非常感謝，import pandas as pdfoo = pd.DataFrame([['A',1,2,4],['B',3,4,2],['C',5,6,1]], columns=('a', 'x', 'y','z'))foo.set_index('a')Out[1]:a x y zA 1 2 4B 3 4 2C 5 6 1對于每一行，檢查列值，如果值 <= 2，則記錄列名稱。對于上面的例子，輸出應(yīng)該是：Output[2]: a 1 2 3A x y B zC z
2 回答
吃雞游戲
TA貢獻(xiàn)1829條經(jīng)驗 獲得超7個贊
使用melt，然后過濾值，然后使用值assign的新鍵cumcount
df=foo.melt('apple').\
loc[lambda x : x.value<=2].\
assign(value=lambda x : x.groupby('apple').cumcount()+1)
df.pivot('apple','value','variable')
Out[56]:
value 1 2
apple
A x y
B z NaN
C z NaN
大話西游666
TA貢獻(xiàn)1817條經(jīng)驗 獲得超14個贊
這是使用列標(biāo)簽mul替換值的一種方法<= 2。然后排序以在每行的末尾放置空字符串。
import pandas as pd
foo = pd.DataFrame([['A',1,2,4],['B',3,4,2],['C',5,6,1]], columns=('apple', 'x', 'y','z'))
arr = (foo.iloc[:, 1:] <= 2).mul(foo.columns[1:].to_series(), axis=1)
arr_sorted = arr.apply(sorted, key=bool, reverse=True, axis=1).values
foo.iloc[:, 1:] = pd.DataFrame(arr_sorted.tolist(), columns=foo.columns[1:])
print(foo)
apple x y z
0 A x y
1 B z
2 C z
添加回答
舉報
0/150
提交
取消