如何在python

时间:2019-06-15 04:57:54

标签: python-3.x

我想对列表中的每个笛卡尔集合进行求和

我将对每个输入进行平方运算,然后从嵌套列表中创建笛卡尔集,并将追加到新列表中。

from itertools import product
n, k = map(int,input().split())
list1 = [[pow(int(l),2) for l in input().split()] for i in range (n)]
list2 = []
for i in product(*list1):
    list2.append(i)

有预定义的功能吗?

1 个答案:

答案 0 :(得分:1)

我认为您想要对产品的每个元组求和:

for i in product(*list1):
    list2.append(sum(i))