I have 2 numpy arrays:
a = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]]) b = np.array([0, 1, 2])
I want to use b as in