使用.argsort来获取基础数组的索引.切片值和列索引以获取所需的所有信息.我们将创建一个MultiIndex,以便我们可以将列标题和值存储在同一个DataFrame中.第一级将是您的原始索引.
例:
import pandas as pd
import numpy as np
np.random.seed(1)
df = pd.DataFrame(np.random.randint(1,100000, (1739, 26)))
df.columns = list('ABCDEFGHIJKLMNOPQRSTUVWXYZ')
N = 7 # 150 in your case
idx = np.argsort(df.values, 1)[:, 0:N]
pd.concat([pd.DataFrame(df.values.take(idx), index=df.index),
pd.DataFrame(df.columns[idx], index=df.index)],
keys=['Value', 'Columns']).sort_index(level=1)
输出:
0 1 2 3 4 5 6
Columns 0 C K U V I G P
Value 0 5193 7752 8445 19947 20610 21441 21759
Columns 1 R J W C B D G
Value 1 432 3607 16278 17138 19434 26104 33879
Columns 2 K S E F M O U
Value 2 16 1047 1845 9553 12314 13784 19432
Columns 3 K Q A S X W G
Value 3 244 5272 10836 13682 29237 33230 34448
Columns 4 K T L U C D M
Value 4 9765 11275 13160 22808 30870 33484 42760
...
Columns 1736 I L U W B X N
Value 1736 7099 7922 11047 12613 15502 18409 21576
Columns 1737 J K E V O M S
Value 1737 3000 5538 7933 13777 17310 22640 24750
Columns 1738 O U A X E I F
Value 1738 5118 5775 10267 11320 17659 30055 30702