반응형
NumPy 배열에서 0과 같은 원소의 인덱스 찾기
NumPy에는 객체 nonzero()
에서 0이 아닌 요소의 인덱스를 식별 하는 효율적인 함수 / 방법 이 ndarray
있습니다. 요소의 인덱스를 얻을 수있는 가장 효율적인 방법은 무엇입니까 할 0의 값을 갖는다는?
numpy.where () 는 내가 가장 좋아하는 것입니다.
>>> x = numpy.array([1,0,2,0,3,0,4,5,6,7,8])
>>> numpy.where(x == 0)[0]
array([1, 3, 5])
있다 np.argwhere
,
import numpy as np
arr = np.array([[1,2,3], [0, 1, 0], [7, 0, 2]])
np.argwhere(arr == 0)
찾은 모든 인덱스를 행으로 반환합니다.
array([[1, 0], # Indices of the first zero
[1, 2], # Indices of the second zero
[2, 1]], # Indices of the third zero
dtype=int64)
다음을 사용하여 스칼라 조건을 검색 할 수 있습니다.
>>> a = np.asarray([0,1,2,3,4])
>>> a == 0 # or whatver
array([ True, False, False, False, False], dtype=bool)
배열을 조건의 부울 마스크로 되돌려줍니다.
nonzero()
조건의 부울 마스크에서 사용하여 사용할 수도 있습니다 False
. 또한 일종의 0 이기 때문 입니다.
>>> x = numpy.array([1,0,2,0,3,0,4,5,6,7,8])
>>> x==0
array([False, True, False, True, False, True, False, False, False, False, False], dtype=bool)
>>> numpy.nonzero(x==0)[0]
array([1, 3, 5])
그것은 mtrw
방법 과 정확히 동일 하지만, 질문과 더 관련이 있습니다.)
1 차원 배열로 작업하는 경우 구문 설탕이 있습니다.
>>> x = numpy.array([1,0,2,0,3,0,4,5,6,7,8])
>>> numpy.flatnonzero(x == 0)
array([1, 3, 5])
import numpy as np
x = np.array([1,0,2,3,6])
non_zero_arr = np.extract(x>0,x)
min_index = np.amin(non_zero_arr)
min_value = np.argmin(non_zero_arr)
나는 다음과 같이 할 것입니다 :
>>> x = np.array([[1,0,0], [0,2,0], [1,1,0]])
>>> x
array([[1, 0, 0],
[0, 2, 0],
[1, 1, 0]])
>>> np.nonzero(x)
(array([0, 1, 2, 2]), array([0, 1, 0, 1]))
# if you want it in coordinates
>>> x[np.nonzero(x)]
array([1, 2, 1, 1])
>>> np.transpose(np.nonzero(x))
array([[0, 0],
[1, 1],
[2, 0],
[2, 1])
numpy.nonzero를 사용하여 0을 찾을 수 있습니다.
>>> import numpy as np
>>> x = np.array([1,0,2,0,3,0,0,4,0,5,0,6]).reshape(4, 3)
>>> np.nonzero(x==0) # this is what you want
(array([0, 1, 1, 2, 2, 3]), array([1, 0, 2, 0, 2, 1]))
>>> np.nonzero(x)
(array([0, 0, 1, 2, 3, 3]), array([0, 2, 1, 1, 0, 2]))
참고 URL : https://stackoverflow.com/questions/4588628/find-indices-of-elements-equal-to-zero-in-a-numpy-array
반응형
'IT story' 카테고리의 다른 글
“캐치”없이“시도 적으로”수행하는 것이 합리적입니까? (0) | 2020.07.17 |
---|---|
Kotlin에서 빌더 패턴을 구현하는 방법은 무엇입니까? (0) | 2020.07.17 |
JavaScript에서 함수를 주기적으로 호출하는 방법이 있습니까? (0) | 2020.07.17 |
PHP 추상 속성 (0) | 2020.07.17 |
CSS 의사 클래스 입력 : not (disabled) not : [type =“submit”] : focus (0) | 2020.07.17 |