python - 如何遍历字典中的每对项目

Question

我正在制作一个重力模拟器，我需要计算作用在每个物体上的合力。

为了做到这一点，我需要遍历字典中的每一对物体(id: instance of Body class)并获得这两个物体之间的引力。然后，我将所有的力相加并得到结果。

但是，如何在 Python 中只对字典中的每对项目进行一次迭代？如果将天体保存在一个列表中，那就很简单了：

for i in range(len(bodies)):
    for j in range(len(bodies) - i - 1):
        k = j - i + 1
        b1 = bodies[i]
        b2 = bodies[k]

Answer 1

values()和itertools'combinations非常适合这个用例。

from itertools import combinations
for a, b in combinations(bodies.values(), 2):
    print a, b

Answer 2

您正在寻找itertools.combinations()：

一个例子：

In [76]: lis=['a','b','c','d']  #consider these as your dictionary items

In [77]: [x for x in combinations(lis,2)]
Out[77]: [('a', 'b'), ('a', 'c'), ('a', 'd'), ('b', 'c'), ('b', 'd'), ('c', 'd')]

Answer 3

该itertools模块提供了一个很好的combinations方法，您可以使用：

from itertools import combinations

bodies = {}
# add bodies

for a,b in combinations(bodies.values(), 2):
    # a and b are a pair of bodies. do stuff
    pass

顺便说一句，即使您使用列表，这仍然有效：

from itertools import combinations

bodies = []
# add bodies

for a,b in combinations(bodies, 2):
    pass