Change Array Dimensions Python