沿额外维度展开python矩阵

时间:2014-06-17 13:51:13

标签: python arrays matrix-multiplication

我有一个2维3x3数组,例如:

(4,5,6

 8, 10, 12

 12,15,18 )

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

1 个答案:

答案 0 :(得分:0)

这是我能提出的最短代码(不是最优化的):

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]