numpy的乘以不同的形狀

Question

具有兩個陣列是這樣

x = [a,b] 
y = [p,q,r] 

我需要一起相乘這對一個產品c這應該是這樣的，

c = [a*p, a*q, a*r, b*p, b*q, b*r] 

然而x*y提供了以下錯誤，

ValueError: operands could not be broadcast together with shapes (2,) (3,) 

我可以這樣做，

for i in range(len(x)): 
    for t in range(len(y)): 
     c.append(x[i] * y[t] 

但真正的長度我x和y是相當大的有啥做出這樣的乘法沒有循環的最有效方式。

Answer 1

可以使用NumPy broadcasting爲x和y之間的成對的elementwise乘法，然後用.ravel()壓平，像這樣 -

(x[:,None]*y).ravel() 

或者使用outer product然後壓平 -

np.outer(x,y).ravel() 

Answer 2

使用NumPy的點。 ..

>>> import numpy as np 
>>> a=np.arange(1,3)# [1,2] 
>>> b=np.arange(1,4)# [1,2,3] 
>>> np.dot(a[:,None],b[None]) 
array([[1, 2, 3], 
     [2, 4, 6]]) 
>>> np.dot(a[:,None],b[None]).ravel() 
array([1, 2, 3, 2, 4, 6]) 
>>>