python 中的最小值 itertools.product

Question

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]

在上面的代码中，有没有办法只return每个a的a + b的最小值？所以输出应该是：

[50.1,60.1,70.1]

请注意这是一个玩具示例，在真实示例中，A 和 B 是经纬度值列表，我计算它们之间的大圆距离

Answer 1

是的，首先selectB的最小值，然后将其添加到A的每个元素。

b = min(B)
print([a + b for a in A])

Answer 2

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 中的最小值 itertools.product

Minimum of itertools.product in python

python

itertools