numpy 一维数组到 numpy 数组答案

【问题标题】：numpy 1D array to numpy arraysnumpy 一维数组到 numpy 数组
【发布时间】：2013-10-24 13:10:06
【问题描述】：

我想在 Python 中读取一维 numpy 数组并生成另外两个 numpy 数组：

例如：

a = numpy.array([1,2,numpy.nan,4])

会给

[1,2,0,4]
[1,1,0,1]

在 python 中最有效的方法是什么？

谢谢

【问题讨论】：

【解决方案1】：

要将nan 替换为0，请使用numpy.nan_to_num：

>>> a = numpy.array([1,2,numpy.nan,4])
>>> numpy.nan_to_num(a)
array([ 1.,  2.,  0.,  4.])

使用numpy.isnan 将nan 转换为True，非nan 数字转换为False。然后从1 中减去它们。

>>> numpy.isnan(a)
array([False, False,  True, False], dtype=bool)
>>> 1 - numpy.isnan(a)
array([ 1.,  1.,  0.,  1.])

【讨论】：

【解决方案2】：

第一个：

numpy.nan_to_num(a)

第二个：

numpy.invert(numpy.isnan(a)).astype(int)

【讨论】：

【解决方案3】：

要将 NaN 转换为零，请使用：

numpy.nan_to_num(a)

要将非 NaN 设置为 1，将 NaN 设置为 0，请尝试：

numpy.isfinite(a)*1

【讨论】：