元组迭代函数

Question

我有以下问题： 我想遍历给定列表并返回所有可能的笛卡尔积之和的最小值：

from itertools import product

z = ((1, 2),(2, 3))

def zmin(tup):
    return min(sum(a*a for a in s) for s in product(tup))

zmin(z) --> ERROR MESSAGE

正确答案应该是这样的：

1*1 + 2*2 =  5
1*1 + 3*3 = 10
2*2 + 2*2 =  8
2*2 + 3*3 = 13

使得zmin(z)的返回值=5

你知道我做错了什么吗？ 还有没有颠簸或类似的有效方法？

Answer 1

我自己找到了一个解决方案： 我为返回值添加了*args：

def zmin(tup):
    return min(sum(a*a for a in s) for s in product(*tup))