查找Pandas中軸上N個最大值的索引

Question

我知道有一個方法.argmax（）返回軸上最大值的索引。

但是，如果我們想要獲得軸上10個最高值的索引呢？

怎么可以實現呢？

例如：

data = pd.DataFrame(np.random.random_sample((50, 40)))

Answer 1

例如，IIUC，如果你想獲得列col的前10個最大數字的索引：

data[col].nlargest(10).index

Answer 2

你可以使用argsort ：

s = pd.Series(np.random.permutation(30))
sorted_indices = s.argsort()
top_10 = sorted_indices[sorted_indices < 10]
print(top_10)

輸出：

3     9
4     1
6     0
8     7
13    4
14    2
15    3
19    8
20    5
24    6
dtype: int64

Answer 3

試一試。 這將獲取一行中的10個最大值並將它們放入數據幀中。

import pandas as pd
import numpy as np

df = pd.DataFrame(np.random.random_sample((50, 40)))
df2 = pd.DataFrame(np.sort(df.values)[:,-10:])