從Numpy中的2個向量形成矩陣，重復1個向量

Question

使用numpy數組我想以最經濟的方式創建這樣一個矩陣：給定

from numpy import array
a = array(a1,a2,a3,...,an)
b = array(b1,...,bm)

應處理為矩陣M：

M = array([[a1,a2,b1,...,an],
           ...           ...,
           [a1,a2,bm,...,an]]

我知道numpy數組的廣播方法，但無法找到一個好方法。 任何幫助將非常感激，

歡呼，羅布

Answer 1

您可以使用numpy.resize上a ，然后再加入b在使用所需的指標的項目numpy.insert重新大小的陣列上：

In [101]: a = np.arange(1, 4)

In [102]: b = np.arange(4, 6)                                           

In [103]: np.insert(np.resize(a, (b.shape[0], a.shape[0])), 2, b, axis=1)                                                                       
Out[103]: 
array([[1, 2, 4, 3],                                                    
       [1, 2, 5, 3]])  

Answer 2

您可以使用numpy.tile和numpy.hstack函數的組合。

M = numpy.repeat(numpy.hstack(a, b), (N,1))

不過，我不確定我是否理解你的目標矩陣。