배열 내의 조건 확인

Question

배열 내부의 조건을 확인하고 조건이 충족되는 위치에서 작업을 수행하고 싶습니다. 예를 들어,이 코드 조각 작업을 수행합니다

res = somefunction(x) 
for i in range(x.shape[0]): 
    for j in range(x.shape[1]): 
     if not 6 < res[i,j] < 18: 
      x[i,j] = float('nan') 

을하지만 빠른 (짧은) 방법은 어쩌면 이런 일이 될 것이라고 생각 :

x[not 6 < somefunction(x) < 18] = float('nan') 

그러나 파이썬은 조건을 확인하는 오류를 제공 요소 이상으로 배열에서는 작동하지 않습니다. 코드를 더 빠르게 만들 수있는 방법이 있습니까?

Answer 1

not이나 연결된 비교를 브로드 캐스트하기 위해 구현할 수 없으므로 배열과의 연결 비교는 사용할 수 없습니다. NumPy와 부울 배열에 부울 연산을위한 비트 연산자를 사용하기 때문에

분할이 개 비교에 체인 비교 및 ​​~를 사용 & 대신 not 및 and :

x[~((6 < res) & (res < 18))] = numpy.nan