插入1-d numpy的陣列在現有2- d陣列

Question

列我有一個2-d 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]]) 

和一個1-d陣列。

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

是否有插入b到一個作爲新的第一列一個優雅的方式？ 這樣

>>> 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]])