Useful fragments of numpy code

numpy-fragment.md

Selection

where

>>> import numpy as np
>>> a = np.array([[7, 4, 3, 7],
...  [7, 2, 5, 4],
...  [1, 7, 5, 1]])
>>> for (i, j) in zip(*np.where(a % 2 == 0)):
...     print("[{0}, {1}]: {2}".format(i, j, a[i, j]))
... 
[0, 1]: 4
[1, 1]: 2
[1, 3]: 4