3

我有一个 Pandas 数据框，每列都有一个概率，我有 10 列。行代表一条记录。我想选择前 3 个概率并返回该列名（有点像 argmax）。

由于我有大量的行，我不想希望使用循环。有人可以帮我这样做吗？

DataFrame 如下所示：

    col0    col1    col2    col3    col4    col5    col6    col7    col8    col9    
0   0.089659    0.0 0.0 0.0 0.228481    0.000000    0.0 0.575523    0.0 0.089667    
1   0.000000    0.0 0.0 0.0 0.000000    0.422125    0.0 0.516324    0.0 0.000000    
2   0.000000    0.0 0.0 0.0 0.000000    0.304416    0.0 0.659216    0.0 0.000000    
3   0.598459    0.0 0.0 0.0 0.100792    0.082668    0.0 0.144577    0.0 0.053502    
4   0.100000    0.1 0.1 0.1 0.100000    0.100000    0.1 0.100000    0.1 0.100000
4

2 回答 2

2

给定 df

df
Out[145]: 
       col0  col1  col2  col3      col4      col5  col6      col7  col8  \
0  0.089659   0.0   0.0   0.0  0.228481  0.000000   0.0  0.575523   0.0   
1  0.000000   0.0   0.0   0.0  0.000000  0.422125   0.0  0.516324   0.0   
2  0.000000   0.0   0.0   0.0  0.000000  0.304416   0.0  0.659216   0.0   
3  0.598459   0.0   0.0   0.0  0.100792  0.082668   0.0  0.144577   0.0   
4  0.100000   0.1   0.1   0.1  0.100000  0.100000   0.1  0.100000   0.1   

       col9  
0  0.089667  
1  0.000000  
2  0.000000  
3  0.053502  
4  0.100000

您可以使用nlargest：

import numpy as np

x=pd.DataFrame(df).T

rslt = pd.DataFrame(np.zeros((0,3)), columns=['top1','top2','top3'])
for i in x.columns:
    df1row = pd.DataFrame(x.nlargest(3, i).index.tolist(), index=['top1','top2','top3']).T
    rslt = pd.concat([rslt, df1row], axis=0)

print rslt

   top1  top2  top3
0  col7  col4  col9
0  col7  col5  col0
0  col7  col5  col0
0  col0  col7  col4
0  col0  col1  col2
于 2016-06-05T03:51:33.353 回答
0

您应该查看nlargest。这可能足以让您继续前进：

df.T.nlargest(3, df.T.columns).T
于 2016-06-05T03:45:41.637 回答