如何获取 numpy 二维数组中非空元素的位置？

Question

我创建了一个这样的数组， a = np.array([[1,2,3,4],[5,6,np.NaN,8]]) 它会像这样显示现在我想获取数组中空值的位置。任何人都可以为此推荐解决方案吗？

Answer 1

你可以用np.argwhere与np.isnan：

import numpy as np

a = np.array([[1,2,3,4],[5,6,np.NaN,8]]) 

print(np.argwhere(np.isnan(a)))

# [[1 2]]