numpy.nonzero¶

numpy.nonzero 함수는 0이 아닌 성분의 인덱스를 반환합니다.

예제1¶

import numpy as np

a = np.array([1, 2, 0, 3, 0, 4])

print(np.nonzero(a))
print(a[np.nonzero(a)])

(array([0, 1, 3, 5], dtype=int64),)
[1 2 3 4]

값을 출력하려면, a[np.nonzero(a)]와 같이 사용하면 됩니다.

import numpy as np

a = np.array([[1,0,0], [0,2,0], [1,1,0]])

print(np.nonzero(a))
print(np.transpose(np.nonzero(a)))

(array([0, 1, 2, 2], dtype=int64), array([0, 1, 0, 1], dtype=int64))
[[0 0]
 [1 1]
 [2 0]
 [2 1]]