在Python中生成矩阵中每行一个元素的所有总和

2024-05-15 • 问答

假设我有一个 4 行 x 3 列的矩阵。例如：

我想生成所有可能总和的列表，通过每行仅取 1 个元素来计算总和。例如：

1 + 4 + 7 + 10 = 22
3 + 5 + 7 + 11 = 26
...

我怎么能在 Python 中做到这一点？

执行 Cartesion 乘积，将该乘积的每个结果中的值相加。您可以使用 itertools，因为它提供了 Cartesion 乘积的迭代器：

import itertools

m=[[ 1,2,3],[ 4,5,6],[ 7,8,9],[10,11,12]]

res = map(sum,itertools.product(*m))

print(list(res))

本文链接：https://www.f2er.com/4163.html