파이썬 기존 값보다 큰 값의 첫 번째 Numpy

numpy에 1D 배열이 있고 값이 numpy 배열의 값을 초과하는 인덱스의 위치를 ​​찾고 싶습니다.

예 :

aa = range(-10,10)

aa 에서 위치를 찾습니다. 여기서 5 값이 초과됩니다.

 

해결 방법

 

이것은 조금 더 빠르며 더 좋아 보입니다.

np.argmax(aa>5)

In [2]: N = 10000

In [3]: aa = np.arange(-N,N)

In [4]: timeit np.argmax(aa>N/2)
100000 loops, best of 3: 52.3 us per loop

In [5]: timeit np.where(aa>N/2)[0][0]
10000 loops, best of 3: 141 us per loop

In [6]: timeit np.nonzero(aa>N/2)[0][0]
10000 loops, best of 3: 142 us per loop

 

참조 페이지 https://stackoverflow.com/questions/16243955