I have a 3D numpy array of shape (1240467,10,19) which I want to slice based on values in a dictionary. The value for a key in a dictionary denotes the index (for the first