Skip to content

Instantly share code, notes, and snippets.

Show Gist options
  • Save tkamishima/7d11fea6c78e90477c6810479e194ec6 to your computer and use it in GitHub Desktop.
Save tkamishima/7d11fea6c78e90477c6810479e194ec6 to your computer and use it in GitHub Desktop.
iterate arrays that is composed of the last two dimensions
In [58]: a = np.arange(72).reshape(3, 2, 3, 4)
In [59]: for index in np.ndindex(*(list(a.shape)[:-2])):
...: print(a[index])
...:
[[ 0 1 2 3]
[ 4 5 6 7]
[ 8 9 10 11]]
[[12 13 14 15]
[16 17 18 19]
[20 21 22 23]]
[[24 25 26 27]
[28 29 30 31]
[32 33 34 35]]
[[36 37 38 39]
[40 41 42 43]
[44 45 46 47]]
[[48 49 50 51]
[52 53 54 55]
[56 57 58 59]]
[[60 61 62 63]
[64 65 66 67]
[68 69 70 71]]
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment