python - 如何使用 Python 对列表中的单词进行笛卡尔积

Question

事实证明我正在寻找组合而不是笛卡尔积。

<小时/>

我需要对几个单词(所有单词都在一个列表中)进行笛卡尔积。

我的输入看起来像这样:

[[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)。其他部分也是同样的事情..

Answer 1

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))