python - 将一维 NumPy 数组作为列插入现有二维数组中

Question

我有一个2D NumPy数组:

>>> import numpy as np
>>> a = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])
>>> a
array([[1, 2, 3],
       [4, 5, 6],
       [7, 8, 9]])

和一个一维数组:

>>> b = np.arange(3)
>>> b
array([0, 1, 2])

是否有一种优雅的方式将 b 插入 a 作为新的第一列？

这样:

>>> a
array([[0, 1, 2, 3],
       [1, 4, 5, 6],
       [2, 7, 8, 9]])

Answer 1

您可以使用column_stack()

In [256]: np.column_stack((b, a))
Out[256]:
array([[0, 1, 2, 3],
       [1, 4, 5, 6],
       [2, 7, 8, 9]])