(Python/Pandas) 排序值時的數據透視表

Question

user_id_2     1        2          3         4         5    
user_id_1                                                                      
   1      0.000000  0.707107  0.388075  0.668153  0.559017  
   2      0.707107  0.000000  0.504916  0.491354  0.632456  
>> 3      0.388075  0.504916  0.000000  0.373383  0.225618 <<
   4      0.668153  0.491354  0.373383  0.000000  0.448211  
   5      0.559017  0.632456  0.225618  0.448211  0.000000  

這是一些計算的結果。 我想知道我是否可以選擇一行並對列中的最高值進行排名。

例如。 選擇行user_id_1(3)

user_id_1      user_id_2
    3              2
    3              1
    3              4
    3              5
    3              3

Answer 1

按DataFrame.loc所選行的降序使用Series.argsort ，獲取列名的新順序並通過構造函數創建新的 DataFrame ：

val = 3
df = pd.DataFrame({'user_id_1':val, 
                   'user_id_2':df.columns[(-df.loc[val]).argsort()]})

print (df)
   user_id_1 user_id_2
0          3         2
1          3         1
2          3         4
3          3         5
4          3         3

或者，如果只想訂購添加1 ：

val = 3
df = pd.DataFrame({'user_id_1':val, 
                   'user_id_2':(-df.loc[val]).argsort() + 1})

print (df)
   user_id_1  user_id_2
1          3          2
2          3          1
3          3          4
4          3          5
5          3          3