numpy.where가 반환하는 값의 수를 찾음

Question

numpy.where가 true로 평가 된 장소의 수를 찾고 싶습니다. 다음 솔루션이 작동하지만 꽤 못 생깁니다.

b = np.where(a < 5) 
num = (b[0]).shape[0] 

결과 배열을 사용하여 작업을 수행하기 전에 num> 0인지 확인해야하는 언어에서 왔습니다. num을 찾는 것보다 더 우아한 방법이 있습니까? 아니면 num을 찾는 것보다 더 Pythonic 한 솔루션입니까?

(IDL에 익숙한 사람들을 위해, 나는 간단한 b = where(a lt 5, num)을 복제하는 것을 시도하고있다.)

Answer 1

In [1]: import numpy as np 

In [2]: arr = np.arange(10) 

In [3]: np.count_nonzero(arr < 5) 
Out[3]: 5 

또는

In [4]: np.sum(arr < 5) 
Out[4]: 5 

---

을 어쨌든 b = np.where(arr < 5)[0]을 정의해야하는 경우, len(b)를 사용 또는 b.size (len()은 조금 더 빨라지지만 성능 측면에서는 거의 동일합니다).

Answer 2

where 문 조건이 결과가 False로 평가되지 않음을 의미하는 경우에만 작동합니다. 그러나 그것은 쉽게 : 그래서

import numpy as np 
a = np.random.randint(1,11,100) 
a_5 = a[np.where(a>5)] 
a_10 = a[np.where(a>10)] 
a_5.any() #True 
a_10.any() #False 

당신은 물론 당신이 할당하기 전에이 시도 할 수하고자하는 경우 :

a[np.where(a>5)].any() #True 
a[np.where(a>10)].any() #False 

Answer 3

또 다른 방법은 당신이 더 당신도 할 수 그것을 좋아한다면

In [14]: a = np.arange(100) 
In [15]: np.where(a > 1000)[0].size > 0 
Out[15]: False 
In [16]: np.where(a > 10)[0].size > 0 
Out[16]: True