將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