Конкатенация np.arrays python

Question

Как мне поступить, если:

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

b = np.array([0,1])

Я ищу объединение a и b, поэтому результат будет:

np.array([1,2,3,0],[5,6,7,1])

Большое спасибо

Answer 1

Использование numpy трансляция с concatenate

np.concatenate([a,b[:,None]],1)
Out[1053]: 
array([[1, 2, 3, 0],
       [5, 6, 7, 1]])

Answer 2

Более умный способ сделать это - избежать вещания и использовать функцию, предназначенную для этого: numpy.column_stack:

np.column_stack([a, b])

array([[1, 2, 3, 0],
       [5, 6, 7, 1]])

Answer 3

Проблема заключается в объединении a по горизонтали с b в качестве вектора столбца.

<concat>( |1 2 3|, |0| )
          |5 6 7|  |1|

Объединение может быть выполнено с использованием np.hstack и b можно преобразовать в вектор-столбец, добавив новую ось:

>>> np.hstack([a, b[:, np.newaxis]])
array([[1, 2, 3, 0],
       [5, 6, 7, 1]])