Python Numpy Change Array Dimensions Matlab Subplot