我有一個 15 列的 df：df.columns：
0 class
1 name
2 location
3 income
4 edu_level
--
14 marital_status
經過一些轉換后，我得到了一個numpy.ndarray名為 (15,3) 的形狀loads：
0.52 0.33 0.09
0.20 0.53 0.23
0.60 0.28 0.23
0.13 0.45 0.41
0.49 0.9
so on so on so on
因此，3 列有 15 個值。
我需要做什么：
我想df column從第一列中獲取loads大于 0.50的值的名稱
loads對于此示例，與值高于 0.5的第一列相關的 df 列應回傳：
0 Class
2 Location
的第二列相同loads，應回傳：
1 name
3 income
4 edu_level
和第 3 列的邏輯相同loads。
我設法以loads他們需要的方式獲得了 numarray，但我在最后一部分中度過了一段糟糕的時光。我知道我可以簡單地手動選擇列，但是當 df 具有超過 15 個功能時，這將是一項艱巨的任務。
任何人都可以幫助我嗎？
uj5u.com熱心網友回復：
給定您的閾值，您可以創建一個布爾陣列以過濾 df.columns：
threshold = .5
for j in range(loads.shape[1]):
print(df.columms[loads[:,j]>threshold])
轉載請註明出處，本文鏈接：https://www.uj5u.com/ruanti/448411.html
標籤：python-3.x 熊猫 麻木的