python - 使用 numpy.ma 屏蔽空白值

Question

我无法获取 numpy.ma处理我的数据。我确定我以前用过它来屏蔽空白值，但不知道如何使用。这是显示我遇到的问题的代码片段。

import numpy as np
import numpy.ma as ma

x = np.array([[0.0, 1.1, '', 2.2, ''],[3.3,'', 4.4, '', 5.5]])
for index, value in np.ndenumerate(x):
    if value == '':
        x[index] = None
x = ma.masked_values(x, None)
print x

打印:

[['0' '1' 'N' '2' 'N']
 ['3' 'N' '4' 'N' '5']]

我想做的是获取一个掩码的 float 数组，其中任何缺失的值都被掩码掉。决赛print应该产生:

[[0.0 1.0 -- 2.0 --]
 [3.0 -- 4.0 -- 5.0]]

Answer 1

如果用 NaN 替换空字符串，则以下内容将起作用:

In [19]: x = np.array([[0.0, 1.1, np.nan, 2.2, np.nan],[3.3,np.nan, 4.4, np.nan, 5.5]])

In [20]: m = ma.masked_array(x, np.isnan(x))

In [21]: print(m)
[[0.0 1.1 -- 2.2 --]
 [3.3 -- 4.4 -- 5.5]]

将 '' 更改为 np.nan 的最简单位置是在将它们放入 np.array 之前。