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:
commit
27179d38a7
@ -2135,7 +2135,7 @@
|
||||
"B = np.random.randint(0,5,(2,2))\n",
|
||||
"\n",
|
||||
"C = (A[..., np.newaxis, np.newaxis] == B)\n",
|
||||
"rows = (C.sum(axis=(1,2,3)) >= B.shape[1]).nonzero()[0]\n",
|
||||
"rows = np.where(C.any((3,1)).all(1))[0]\n",
|
||||
"print(rows)"
|
||||
]
|
||||
},
|
||||
|
@ -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)
|
||||
```
|
||||
|
||||
|
Loading…
x
Reference in New Issue
Block a user