事实证明我正在寻找组合而不是笛卡尔积。
<小时/>我需要对几个单词(所有单词都在一个列表中)进行笛卡尔积。
我的输入看起来像这样:
[[id, [word1,word2,word3]],[id2,[word4,word5,word6]]]
需要根据每个列表列表中第一个索引中的单词进行乘积。含义 - 在 word1、word2 和 word3 之间,在 word4、word5 和 word6 之间,依此类推。
到目前为止我的代码:
for row in x:
row[1] = list(itertools.product(*row[1])
问题是脚本对每个字母而不是每个单词进行乘积。
我得到类似:(w,w,w)、(w,o,r)、(w,o,d)
等
预期输出为:(word1,word2), (word1,word3), (word2,word3)
。其他部分也是同样的事情..
最佳答案
itertools.product
将多个可迭代对象作为输入。笛卡尔积是在给定的可迭代之间获取的。例如。 itertools.product([1, 2], [3, 4])
给出 [(1, 3), (1, 4), (2, 3), (2, 4 )]
。
话虽如此,您实际上要调用的是 itertools.combinations
for row in x:
row[1] = list(itertools.combinations(row[1], 2))
关于python - 如何使用 Python 对列表中的单词进行笛卡尔积,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/33066034/