Cacher is the code snippet organizer for pro developers

#!/usr/bin/env python

import numpy as np


a = np.arange(2 * 3 * 4).reshape((2, 3, 4))
b = np.arange(3) % 2

# print(a[b])
# print(a[b].shape)
# a[b][0, 0] == a[b[0, 0], :, :]
print a[b, b].shape
a[b, b][0] == a[b[0], b[0]]
print a[b, :, b].shape 
print a.transpose(0, 2, 1)[b, b][0] == a[b[0], :, b[0]]

print a[np.array([True, False]), [1, 0]]
print (a[np.array([True, False])])[0][[1, 0]]
print a[0].shape
print a[np.array([True, False]), 1:3]