(Python / Pandas) Сводная таблица при сортировке значений - PullRequest
Новая
       3

(Python / Pandas) Сводная таблица при сортировке значений

1 голос
/ 03 февраля 2020 
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

1 Ответ

0 голосов
/ 03 февраля 2020

Используйте Series.argsort в в порядке убывания выбранной строки на DataFrame.loc, получите новый порядок имен столбцов и создайте новый 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
...