python - 从三维数组中删除 NaN 行

Question

如何使用索引从下面的数组中删除 NaN 行(因为我需要从不同的数组中删除相同的行。

array([[[nan,  0.,  0.,  0.],
        [ 0.,  0.,  0.,  0.],
        [ 0.,  0.,  0.,  0.]],

       [[ 0.,  0.,  0.,  0.],
        [ 0., nan,  0.,  0.],
        [ 0.,  0.,  0.,  0.]]])

我使用命令获取要删除的行的索引

a[np.isnan(a).any(axis=2)]

但是使用我通常在二维数组上使用的内容并没有产生预期的结果，丢失了数组结构。

a[~np.isnan(a).any(axis=2)]
array([[0., 0., 0., 0.],
       [0., 0., 0., 0.],
       [0., 0., 0., 0.],
       [0., 0., 0., 0.]])

如何使用从第一个命令获得的索引删除我想要的行？

Answer 1

你需要 reshape :

a[~np.isnan(a).any(axis=2)].reshape(a.shape[0], -1, a.shape[2])

但请注意，每个 2D 子数组的 NaN 行数应该相同才能获得新的 3D 数组。