Merge pull request #33 from ibah/master
Solution to How to find rows of A that contain elements of each row of B
This commit is contained in:
@@ -1108,7 +1108,7 @@ A = np.random.randint(0,5,(8,3))
|
||||
B = np.random.randint(0,5,(2,2))
|
||||
|
||||
C = (A[..., np.newaxis, np.newaxis] == B)
|
||||
rows = (C.sum(axis=(1,2,3)) >= B.shape[1]).nonzero()[0]
|
||||
rows = np.where(C.any((3,1)).all(1))[0]
|
||||
print(rows)
|
||||
```
|
||||
|
||||
|
||||
Reference in New Issue
Block a user