创建包含列表中所有单词的bigram列表

Question

我试图在一个列表中的所有单词之间创建bigram。我可以用单词+顺流来创造双字组。但我不能把一个词和名单上的其他词结合起来

脚本：

countries = ['france', 'brazil', 'usa', 'uk', 'canada', 'mexico', 'angola']

countries_bigrams = zip(*[countries[i:] for i in range(2)])

print(list(countries_bigrams))

我的输出：

[('france', 'brazil'), ('brazil', 'usa'), ('usa', 'uk'), ('uk', 'canada'), ('canada', 'mexico'), ('mexico', 'angola')]

良好输出：

[('france', 'brazil'), ('france', 'usa'), ('france', 'uk'), ('france', 'canada'), ('france', 'mexico'), ('france', 'angola'), ('brazil', 'france'), ('brazil', 'usa'), ('brazil', 'uk'), ('brazil', 'canada'), ('brazil', 'mexico'), ('brazil', 'angola'), ...]

Answer 1

您需要计算大小2的组合。一个好方法是使用itertools：

import itertools
countries = ['france', 'brazil', 'usa', 'uk', 'canada', 'mexico', 'angola']
list(itertools.combinations(countries, 2))

作为输出：

[('france', 'brazil'),
 ('france', 'usa'),
 ('france', 'uk'),
 ('france', 'canada'),
 ('france', 'mexico'),
 ('france', 'angola'),
 ('brazil', 'usa'),
 ('brazil', 'uk'),
 ('brazil', 'canada'),
 ('brazil', 'mexico'),
 ('brazil', 'angola'),
 ('usa', 'uk'),
 ('usa', 'canada'),
 ('usa', 'mexico'),
 ('usa', 'angola'),
 ('uk', 'canada'),
 ('uk', 'mexico'),
 ('uk', 'angola'),
 ('canada', 'mexico'),
 ('canada', 'angola'),
 ('mexico', 'angola')]

我认为这是一个更快的方法，我比较了我的机器的性能：

%%timeit
list(itertools.combinations(countries, 2))

大约需要1.28µs码

%%timeit
[(c1, c2) for c1 in countries for c2 in countries if c1 != c2]

大约需要4.81µs码

%%timeit
countries_bigrams = [(x, y) for x, y in product(countries, repeat=2) if x != y]

大约5.81µs码