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:
Nicolas P. Rougier 2016-11-28 17:06:45 +01:00 committed by GitHub
commit 27179d38a7
2 changed files with 2 additions and 2 deletions

View File

@ -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)"
]
},

View File

@ -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)
```