Python:如何对两个列表中的元素进行乘法和求和(不使用库)？

Question

我有两个包含数字列表的列表(m1 和 m2)。我正在尝试按元素进行乘法和求和，包括叉积以获得最终列表(结果)，如下所示:

m1 = [[1, 2, 3], [4, 5, 6]]
m2 = [[7, 9, 2], [8, 1, 3]]

[[1*7+2*9+3*2,1*8+2*1+3*3],[4*7+5*9+6*2,4*8+5*1+6*3]]

result = [[31,19],[85,55]]

Answer 1

你可以玩python built-in functions和嵌套列表理解:

>>> [[sum(t*k for t,k in zip(i,j)) for j in m2] for i in m1]
[[31, 19], [85, 55]]

您还可以使用 itertools.product 来查找子列表之间的产品:

>>> from itertools import product
>>> [sum(t*k for t,k in zip(i,j)) for i,j in product(m1,m2)]
[31, 19, 85, 55]