numpy.ndarray去除nan元素

可以使用numpy.isnan()函数找到nan元素的位置，然后使用numpy.delete()函数删除这些元素。具体代码如下：

import numpy as np

a = np.array([1, 2, np.nan, 4, np.nan, 6])
print(a)

# 找到nan元素的位置
nan_index = np.isnan(a)

# 删除nan元素
a = np.delete(a, np.where(nan_index))

print(a)