python numpy 에서 nonzero()의 용법

불 배열 을 직접 아래 표 시 된 대상 이나 원 그룹 아래 표 시 된 대상 에 불 배열 이 있 을 때,모두 nonzero()로 불 배열 을 정수 배열 로 바 꾼 다음 정수 배열 을 사용 하여 아래 표 시 된 연산 을 하 는 것 과 같다.
nonzeros(a)는 배열 a 의 중간 값 이 0 이 아 닌 요소 의 아래 표 시 를 되 돌려 줍 니 다.반환 값 은 a.ndim(배열 a 의 축 수)길이 의 원 그룹 입 니 다.원 그룹의 모든 요 소 는 하나의 정수 배열 이 고 그 값 은 0 이 아 닌 요소 의 아래 표 시 는 대응 축 에 있 는 값 입 니 다.예 를 들 어 1 차원 불 배열 b1,nonzero(b1)는 길이 가 1 인 원 그룹 을 얻 었 는데 이것 은 b1[0]과 b1[2]의 값 이 0(False)이 아니 라 는 것 을 나타 낸다.

>>> b1 = np.array([True, False, True, False])
>>> np.nonzero(b1)
    (array([0, 2]),)

2 차원 배열 b2 에 대해 nonzero(b2)는 길이 가 2 인 원 그룹 을 얻 었 다.그것 의 0 번 째 요 소 는 수조 a 중 값 이 0 이 아 닌 원소 의 0 번 째 축의 하 표 이 고,1 번 째 요 소 는 1 축의 하 표 이기 때문에 아래 의 결 과 를 통 해 b2[0,0],b[0,2]와 b2[1,0 의 값 이 0 이 아니 라 는 것 을 알 수 있다.

>>> b2 = np.array([[True, False, True], [True, False, False]])
>>> np.nonzero(b2)
    (array([0, 0, 1]), array([0, 2, 0]))

불 배열 이 직접 아래 표 시 를 할 때 nonzero()에서 변 환 된 원 그룹 을 아래 표 시 된 대상 으로 사용 하 는 것 과 같 습 니 다.

>>> a = np.arange(3*4*5).reshape(3,4,5)
>>> a[b2]
array([[ 0,  1,  2,  3,  4],
       [10, 11, 12, 13, 14],
       [20, 21, 22, 23, 24]])
>>> a[np.nonzero(b2)]
array([[ 0,  1,  2,  3,  4],
       [10, 11, 12, 13, 14],
       [20, 21, 22, 23, 24]])

아래 표 시 된 대상 이 원 그룹 이 고 그 중 에 불 배열 이 있 을 때 불 배열 을 이유 로 nonzeros()를 전환 한 후의 각 정수 배열 에 해당 합 니 다.

>>> a[1:3, b2]
array([[20, 22, 25],
       [40, 42, 45]])
>>> a[1:3, np.nonzero(b2)[0], np.nonzero(b2)[1]]
array([[20, 22, 25],
       [40, 42, 45]])