我有一个 2D numpy 数组。该数组中的一些值是NaN
。我想使用这个数组执行某些操作。例如考虑数组:
[[ 0. 43. 67. 0. 38.]
[ 100. 86. 96. 100. 94.]
[ 76. 79. 83. 89. 56.]
[ 88. NaN 67. 89. 81.]
[ 94. 79. 67. 89. 69.]
[ 88. 79. 58. 72. 63.]
[ 76. 79. 71. 67. 56.]
[ 71. 71. NaN 56. 100.]]
我试图一次一个地取出每一行,以相反的顺序对其进行排序,以从该行中获取最多 3 个值并取它们的平均值。我尝试的代码是:
# nparr is a 2D numpy array
for entry in nparr:
sortedentry = sorted(entry, reverse=True)
highest_3_values = sortedentry[:3]
avg_highest_3 = float(sum(highest_3_values)) / 3
这不适用于包含以下内容的行:NaN
。我的问题是,有没有一种快速的方法来转换所有NaN
2D numpy 数组中的值为零,这样我在排序和我尝试做的其他事情上就没有问题。
Where A
是你的二维数组:
import numpy as np
A[np.isnan(A)] = 0
功能isnan
生成一个 bool 数组,指示NaN
值是。布尔数组可以用来索引相同形状的数组。把它想象成一个面具。
本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系:hwhale#tublm.com(使用前将#替换为@)