让我们假设一个形状数组,(n,5,2)
其中包含NaN
随机位置的 s,由以下代码生成:
n = 10
arr = np.random.rand(n, 5, 2)
# replace some values by nan
arr = arr.ravel()
index_array = np.arange(arr.size)
np.random.shuffle(index_array)
arr[index_array[:5]] = np.nan
arr = arr.reshape(n, 5, 2)
我怎样才能有效地过滤这个数组,以便只arr[i]
保留那些不包含NaN
s 的 s?然后生成的形状将(m,5,2)
是m<=n
。