Numpy：输入NaN值不是buggy（sklearn）就是不改变数组

Question

目前我正在尝试在一个大的numpy数组中去掉NaN值。如果我应用sklearn中的inputer，它会假设数组中的某些列只包含nan并丢弃它们—尽管只有前几行符合这个假设。你知道吗

我认为这种行为一定是一个bug，因此，我不能使用这个工具（毫不奇怪，我需要转储的列）。但是使用中间值进行实际操作，我得到的是一个完全不变的数组。：（

这是我的代码，x\u train是一个numpy数组：

x_train = x_train.T
for column in x_train:
    median = column[int(len(column)/2)]
    column[column == np.nan] = median
    column[column == np.inf] = 0
    column[column == -np.inf] = 0
x_train = x_train.T

在操作之前和之后打印数组的第一行，最后得到两个相同的行：

[  4.40572853e-01   4.39998817e-01   4.44360730e-01   4.01524000e+02
   4.01524000e+02   4.01524000e+02   4.83419270e-02   4.82160365e-02
   4.91767511e-02              nan              nan              nan
              nan              nan              nan   3.33333333e+01
   3.33333333e+01   3.33333333e+01              nan              nan
              nan]

我错过了什么？我做了很多谷歌搜索，如果答案已经在那里，我一定是在一个完全错误的方向搜索。谢谢你的帮助。你知道吗

Answer 1

列未排序，并且包含您应该使用np.nanmean() 你应该用np.isnan()而不是==。你知道吗

x_train = np.array([[np.inf, 1, 1, np.nan], [2, 2, np.nan, 2,], [10, np.nan, 10, 10]])

print(x_train)

x_train = x_train.T
for column in x_train:
    # median = column[int(len(column)/2)] 
    median = np.nanmedian(column)
    # column[column == np.nan] = median
    column[np.isnan(column)] = median
    column[column == np.inf] = 0
    column[column == -np.inf] = 0
x_train = x_train.T

print(x_train)

印刷品

[[ inf   1.   1.  nan]
 [  2.   2.  nan   2.]
 [ 10.  nan  10.  10.]]

[[  0.    1.    1.    6. ]
 [  2.    2.    5.5   2. ]
 [ 10.    1.5  10.   10. ]]