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

Question

我创建了一个这样的数组， a = np.array([[1,2,3,4],[5,6,np.NaN,8]]) 它会像这样显示

现在我想获取数组中空值的位置。任何人都可以为此推荐解决方案吗？

Answer 1

你可以用np.argwhere与np.isnan：import numpy as npa = np.array([[1,2,3,4],[5,6,np.NaN,8]]) print(np.argwhere(np.isnan(a)))# [[1 2]]