Numpy Array Shape Explained Episodes