将numpy.array a与numpy.array b的前导尺寸相乘

Question

我有两个numpy数组a和b ，其中a的尺寸为1且长度与b的前导尺寸相同，例如，

import numpy
a = numpy.random.rand(5)
b1 = numpy.random.rand(5)
b2 = numpy.random.rand(5, 3, 11)

我想乘每“行” b与相应的条目a ，并获得相同的形状的阵列b 。

就像是

a[:, None, None] * b2

仅在我事先知道b的维数的情况下有效。

Answer 1

你可以用

 (a * b.T).T

如果b是连续的，这将产生一个连续的数组

(a*b.T).flags.contiguous
# False
(a*b.T).T.flags.contiguous
# True