Question

使用Python，我想比较列表中的每个可能的对。

假设我有

my_list = [1,2,3,4]

我想对列表中2个元素的每个组合进行操作（让我们称之为foo）。

最终结果应与

相同

foo(1,1)
foo(1,2)
...
foo(4,3)
foo(4,4)

我的第一个想法是手动在列表中迭代两次，但这似乎不是非常pythonic。

Answer 1

在itertools模块中查看product()。它完全符合您的描述。

import itertools

my_list = [1,2,3,4]
for pair in itertools.product(my_list, repeat=2):
    foo(*pair)

这相当于：

my_list = [1,2,3,4]
for x in my_list:
    for y in my_list:
        foo(x, y)

修改：还有两个非常相似的功能，permutations()和combinations()。为了说明它们的不同之处：

product()生成所有可能的元素配对，包括所有重复项：

1,1  1,2  1,3  1,4
2,1  2,2  2,3  2,4
3,1  3,2  3,3  3,4
4,1  4,2  4,3  4,4

permutations()生成每个唯一元素对的所有唯一排序，从而消除x,x重复项：

 .   1,2  1,3  1,4
2,1   .   2,3  2,4
3,1  3,2   .   3,4
4,1  4,2  4,3   .

最后，combinations()仅按字典顺序生成每个唯一的元素对：

 .   1,2  1,3  1,4
 .    .   2,3  2,4
 .    .    .   3,4
 .    .    .    .

所有这三个功能都是在Python 2.6中引入的。

Answer 2

我遇到了类似的问题并找到了解决方案here。无需导入任何模块即可运行。

假设列表如下：

people = ["Lisa","Pam","Phil","John"]

简化的单行解决方案将如下所示。

所有可能的对，包括重复：

result = [foo(p1, p2) for p1 in people for p2 in people]

所有可能的配对，不包括重复：

result = [foo(p1, p2) for p1 in people for p2 in people if p1 != p2]

唯一配对，订单无关紧要：

result = [foo(people[p1], people[p2]) for p1 in range(len(people)) for p2 in range(p1+1,len(people))]

如果您不想操作但只是为了获得对，删除函数foo并仅使用元组就足够了。

所有可能的对，包括重复：

list_of_pairs = [(p1, p2) for p1 in people for p2 in people]

结果：

('Lisa', 'Lisa')
('Lisa', 'Pam')
('Lisa', 'Phil')
('Lisa', 'John')
('Pam', 'Lisa')
('Pam', 'Pam')
('Pam', 'Phil')
('Pam', 'John')
('Phil', 'Lisa')
('Phil', 'Pam')
('Phil', 'Phil')
('Phil', 'John')
('John', 'Lisa')
('John', 'Pam')
('John', 'Phil')
('John', 'John')

所有可能的配对，不包括重复：

list_of_pairs = [(p1, p2) for p1 in people for p2 in people if p1 != p2]

结果：

('Lisa', 'Pam')
('Lisa', 'Phil')
('Lisa', 'John')
('Pam', 'Lisa')
('Pam', 'Phil')
('Pam', 'John')
('Phil', 'Lisa')
('Phil', 'Pam')
('Phil', 'John')
('John', 'Lisa')
('John', 'Pam')
('John', 'Phil')

唯一配对，订单无关紧要：

list_of_pairs = [(people[p1], people[p2]) for p1 in range(len(people)) for p2 in range(p1+1,len(people))]

结果：

('Lisa', 'Pam')
('Lisa', 'Phil')
('Lisa', 'John')
('Pam', 'Phil')
('Pam', 'John')
('Phil', 'John')

编辑：在简化此解决方案的返工之后，我意识到它与Adam Rosenfield的方法相同。我希望更大的解释能帮助一些人更好地理解它。

Answer 3

如果你只是在调用一个函数，那么你实际上不能做得更好：

for i in my_list:
    for j in my_list:
        foo(i, j)

如果您想收集调用该函数的结果列表，您可以执行以下操作：

[foo(i, j) for i my_list for j in my_list]

会向您返回将foo(i, j)应用于每个可能的对(i, j)的结果列表。

Answer 4

my_list = [1,2,3,4]

pairs=[[x,y] for x in my_list for y in my_list]
print (pairs)

对列表中的每对元素进行操作

4 个答案: