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