沿额外维度展开python矩阵

Question

我有一个2维3x3数组，例如：

(4,5,6

 8, 10, 12

 12,15,18 )

我想将它乘以向量(1,2,3)，这样我最终得到一个3x3x3数组，其中沿着第三维，原始数组的所有元素都乘以1,2或3分别。它是如何在python中做到的？

Answer 1

这是我能提出的最短代码（不是最优化的）：

a = [[1,2,3],[4,5,6],[7,8,9]]
b = [1,2,3]
mult = [[[z*x for z in y] for y in a] for x in b]