2

假设我有一个列表，我想在不考虑顺序的情况下生成所有唯一元素对的列表。一种方法是：

mylist = ['W','X','Y','Z']
for i in xrange(len(mylist)):
    for j in xrange(i+1,len(mylist)):
        print mylist[i],mylist[j]
W X
W Y
W Z
X Y
X Z
Y Z

我想用迭代器做到这一点，我想到了以下几点，即使它不简洁：

import copy
it1 = iter(mylist)
for a in it1:
    it2 = copy.copy(it1)
    for b in it2:
        print a,b

但这甚至行不通。使用迭代器或 zip 等执行此操作的更 Pythonic 和有效的方法是什么？

4

2 回答 2

8

这已经完成并包含在 Python 2.6 的标准库中：

import itertools

mylist = ['W', 'X', 'Y', 'Z']
for pair in itertools.combinations(mylist, 2):
    print pair        # pair is a tuple of 2 elements

对我来说似乎很 Pythonic ;-)

请注意，即使您正在计算大量组合，该combinations()函数也会返回一个迭代器，以便您可以立即开始打印它们。请参阅文档。

此外，您将结果称为列表与其自身之间的笛卡尔积，但这并不完全正确：笛卡尔积将具有 16 个元素 (4x4)。您的输出是其中的一个子集，即只有列表值的 2 元素组合（不允许重复）。

于 2011-03-11T01:17:37.890 回答
0

@Cameron 的回答是正确的。

我只是想指出

for i in range(len(mylist)):
    do_something_to(mylist[i])

非常不符合 Python 风格；如果您的操作是只读的（不必存储回数组），请执行

for i in mylist:
    do_something_to(i)

否则

mylist = [do_something_to(i) for i in mylist]
于 2011-03-11T01:41:56.283 回答