numpy乘以不同形状的数组[重复]

Question

我有一个数组A，形状为(w,h) = 3000,2000 和另一个形状为d = 100的数组B

我想将A的每个值乘以B，得到一个数组C的形式为(w,h,d) = 3000,2000,100的结果

现在我正在使用下面的非常慢的代码，我该如何向量化这个操作？

w,h,d = 3000,2000,100

A = np.ones((w,h))

B = np.arange(d)

C = np.zeros((w,h,d))

for i in xrange(w):
    for j in xrange(h):
        C[i,j] = A[i,j] * B

谢谢

Answer 1

使用 numpy broadcast。

试试这个

from numpy.random import rand
a = rand(4,5)
b = rand(6)
c = a[...,None] * b
print (c.shape)

或者等价

c = a.reshape(4,5,1)*b