gpt4 book ai didi

python - 如何过滤掉包含 NaN 的子数组?

转载 作者:太空宇宙 更新时间:2023-11-03 15:17:24 26 4
gpt4 key购买 nike

让我们假设一个形状为 (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秒?生成的形状将是 (m,5,2)m<=n .

最佳答案

无需 reshape 任何东西:

has_nans = np.isnan(arr).any(axis=(-1,-2))
has_nans
array([False, False, False, True, True, True, False, False, False, True], dtype=bool)

>>> arr = arr[~has_nans]
>>> arr.shape
(6, 5, 2)

旧版本的 numpy 您需要执行以下操作:

has_nans = np.isnan(arr).any(axis=-1).any(axis=-1)

关于python - 如何过滤掉包含 NaN 的子数组?,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/19860262/

26 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com