빠른 ndarray 부울 교차

Question

동일한 모양의 두 개의 3D 부울 배열이 있습니다 (예 : (512,512,512)). 한 배열의 True 영역이 두 번째 배열에 완전히 포함되어 있는지 확인하고 싶습니다.

는 지금 내가 가지고 올 것 중에 최고입니다 :이 작업을 수행하는 빠른 방법은

np.all((A * B) == A) 

있습니까? 불행히도 내부 루프 내부로 들어가서 더 이상 최적화 할 수 없으므로 더 많이 면도 할 수 있습니다.

Answer 1

# 1 접근 : 그들은 부울 배열 인 경우이 AND-ing 곱셈보다 더 좋을 수도 -

((A & B) == A).all() 

접근 방법 # 2 : -

np.count_nonzero(A & B) == np.count_nonzero(A) 

진정한 가치의 계산에 또 다른 런타임 테스트 -

In [225]: A = np.random.rand(512,512,512)>0.5 
    ...: B = np.random.rand(512,512,512)>0.5 

In [226]: %timeit np.all((A * B) == A) 
    ...: %timeit ((A & B) == A).all() 
    ...: %timeit np.count_nonzero(A & B) == np.count_nonzero(A) 
10 loops, best of 3: 170 ms per loop 
10 loops, best of 3: 169 ms per loop 
10 loops, best of 3: 57.7 ms per loop 

---

접근 # 3 :은 기본적으로 해당 B이되지 않은, 어떤 A 사실이 있는지 찾고 numba 수준에서 누전 -이 Short-circuiting와

from numba import njit 

@njit 
def numba_app(a, b): 
    l = len(a) 
    out = True 
    for i in range(l): 
     if a[i] & ~b[i]: 
      out = False 
      break 
    return out 

out = numba_app(A.ravel(), B.ravel()) 

타이밍 방법은 데이터에 따라 다릅니다.