Конкатенация векторов и матриц в виде столбцов

Question

Сейчас я делаю что-то вроде:

In [75]: a = np.arange(7)

In [76]: b = np.arange(3 * 7).reshape([3, 7])

In [77]: np.stack([a, *b], axis=1)
Out[77]:
array([[ 0,  0,  7, 14],
       [ 1,  1,  8, 15],
       [ 2,  2,  9, 16],
       [ 3,  3, 10, 17],
       [ 4,  4, 11, 18],
       [ 5,  5, 12, 19],
       [ 6,  6, 13, 20]])

Есть ли лучшая альтернатива np.stack, где мне не нужно было бы распаковывать b?Мне интересно, какой самый читаемый способ.

Answer 1

Вы можете транспонировать b и затем использовать np.column_stack:

np.column_stack((a, b.T))
#array([[ 0,  0,  7, 14],
#       [ 1,  1,  8, 15],
#       [ 2,  2,  9, 16],
#       [ 3,  3, 10, 17],
#       [ 4,  4, 11, 18],
#       [ 5,  5, 12, 19],
#       [ 6,  6, 13, 20]])