numpy 删除Nan数据

界面

import numpy as np

points_data = np.array([
    [np.nan, 0, np.nan, 0, 0],
    [0, 0, 0, 0, 0],
    [np.nan, np.nan, np.nan, 0, 0],
    [np.nan, np.nan, np.nan, 0, 0],
    [np.nan, np.nan, np.nan, 0, 0],
    [1, 0, 0, np.nan, np.nan],
])
print(np.isnan(points_data)[:, :3])
a = np.delete(points_data, np.where(np.isnan(points_data[:, :3]))[0], axis=0)
print(a)

output:

[[ True False  True]
 [False False False]
 [ True  True  True]
 [ True  True  True]
 [ True  True  True]
 [False False False]]
[[ 0.  0.  0.  0.  0.]
 [ 1.  0.  0. nan nan]]

posted @ 2022-01-25 08:58  不能说的秘密  阅读(1315)  评论(0编辑  收藏  举报