import itertools

A = [50, 60, 70]
B = [0.3, 0.1, 0.5, 0.4]

print [a + b for a, b in itertools.product(A, B)]

>> [50.3,50.1,50.5,50.4,60.3,60.1,60.5,60.4,70.3,70.1,70.5,70.4]

在上面的代码中,是否有方法只返回每个a的最小值a+b?所以输出应该是:
[50.1,60.1,70.1]

请注意,这是一个玩具示例,在实际示例中,a和B是lat和lon值的列表,我计算它们之间的大圆距离

最佳答案

itertools.product并不适合您的大圆任务,因为它将所有(A,B)对集中到一个容器中,但是您希望将具有给定A值的所有对放在一个子容器中,以便找到它们的最小值。所以只要使用一对嵌套循环。如果需要,可以将其作为生成器表达式来计算嵌套在列表理解中的最小值,如下所示:

A = [50, 60, 70]
B = [0.3, 0.1, 0.5, 0.4]

f = lambda x, y: x + y

print [min(f(a,b) for b in B) for a in A]

关于python - python中的itertools.product的最小值,我们在Stack Overflow上找到一个类似的问题:https://stackoverflow.com/questions/29578795/

10-12 21:58