argmin在包含NaN python的数据集中

时间:2016-10-01 14:37:04

标签: python pandas machine-learning

      SGSIN VNVUT CNSHK HKHKG JPOSA
To                                 
MYPKL     1     4     8     9    13
SGSIN   NaN     3     7     8    12
VNVUT   NaN   NaN     3     4     8
CNSHK     1   NaN   NaN     1     5
HKHKG   NaN   NaN   NaN   NaN     3

假设我们使用pandas拥有上述数据集。我想计算第一列上的arg_minimum并忽略NaN。我试过

df[df[0]].idxmin()

但它给出了

nan
但是我没有得到正确的结果。有人能帮我吗?我想要的结果是(在这种情况下)

[0,3]

1 个答案:

答案 0 :(得分:0)

您可以使用numpy的argwhere

import numpy as np

np.argwhere(df['SGSIN'].eq(df['SGSIN'].min()))

array([[0],
       [3]])