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

Question

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

现在，我想获取空值在数组中的位置。有人可以为此建议解决方案吗？

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]]