如何选择行中第二小的值并返回它们各自的列名称?

Aud*_*ott 5 python matrix min dataframe pandas

我有一个数据框,其中有很多列,其中有很多整数值。我想返回行中第二低值的列名称。

我能够返回一行中最低值的列名称,这非常简单:

import pandas as pd

matrix = [(22, 2, 13),
          (9, 1, 5),
          (5, 4, 3),
          (6, 3, 1),
          (1, 2, 20)]

dfObj = pd.DataFrame(matrix, index=list('abcde'), columns=list('xyz'))

minValueIndexObj = dfObj.idxmin(axis=1)
print("min values of row are at following columns :")
print(minValueIndexObj)

Out[]:
min values of row are at following columns :
a    y
b    y
c    z
d    z
e    x
dtype: object
Run Code Online (Sandbox Code Playgroud)

行“a”的列“y”具有最低值。

接下来我需要的是:

2nd min values of row are at following columns :
a    z
b    z
c    y
d    y
e    y
dtype: object
Run Code Online (Sandbox Code Playgroud)

感谢您的支持。

jez*_*ael 3

用于argsort按排序值排列的所有列名称的数组:

a = dfObj.columns.values[np.argsort(dfObj.values)]
print (a)
[['y' 'z' 'x']
 ['y' 'z' 'x']
 ['z' 'y' 'x']
 ['z' 'y' 'x']
 ['x' 'y' 'z']]
Run Code Online (Sandbox Code Playgroud)

然后通过索引选择'columns'并传递给Series构造函数:

print (pd.Series(a[:, 0], index=dfObj.index))
a    y
b    y
c    z
d    z
e    x
dtype: object

print (pd.Series(a[:, 1], index=dfObj.index))
a    z
b    z
c    y
d    y
e    y
dtype: object
Run Code Online (Sandbox Code Playgroud)