Combining multiple 1D arrays returned from a function into a 2D array python,"np.array([a, a, a])"