Добавить размерный массив NumPy

Question

У меня есть два массива, таких как:

a = np.array([1, 2, 3])

b = np.array([101, 102, 103 ])

Я хочу создать новый массив с формой (len (a), 2), такой как

array([[1, 101], [2, 102], [3, 103]])

Как я могу сделать это с NumPy?

Answer 1

Это так называемое column_stack

np.column_stack((a,b))
Out[309]: 
array([[  1, 101],
       [  2, 102],
       [  3, 103]])

Answer 2

В качестве альтернативы:

np.c_[a,b]

также выполнит эту работу.