Как получить расположение ненулевых элементов в массиве 2D?

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