用于查找两个列表的笛卡尔积的 Python 程序

假设我们有两个数据列表 l1 和 l2。我们必须找到这两个列表的笛卡尔积。我们知道，如果两个列表像 (a, b) 和 (c, d)，那么笛卡尔积将是 {(a, c), (a, d), (b, c), (b, d)}。为此，我们将使用 itertools 库并使用该库中的 product() 函数。此函数的返回值是一个迭代器。我们必须通过将输出传递到 list() 构造函数中将其转换为列表。

因此，如果输入为 l1 = [1,5,6] l2 = [1,2,9]，则输出将是 [(1, 1), (1, 2), (1, 9), (5, 1), (5, 2), (5, 9), (6, 1), (6, 2), (6, 9)]

为了解决这个问题，我们将遵循以下步骤 −

• x := product(l1, l2) 获取笛卡尔积的迭代器

• ret := list(x) 将 x 迭代器转换为列表

• 返回 ret

示例

让我们看看下面的实现以便更好地理解

from itertools import product
def solve(l1, l2):
   return list(product(l1, l2))

l1 = [1,5,6]
l2 = [1,2,9]
print(solve(l1, l2))

输入

[1,5,6], [1,2,9]

输出

[(1, 1), (1, 2), (1, 9), (5, 1), (5, 2), (5, 9), (6, 1), (6, 2), (6, 9)]

相关文章

• Python 中通过 K 扩展列表
• Python - 使用交替的 0 和 1 初始化列表
• Python 中元组列表到字典的转换
• Python - 从两个列表中配对，使得成对的元素不相同
• 使用 Python 和 SQLite 的 SQL
• 使用 pytrie 模块在 Python 中进行前缀匹配
• 创建和访问 Python 包
• 如何使用正则表达式和数据类型选择多个 DataFrame 列
• 如何在 Pandas 中查找和过滤重复行?
• 用 Python 编写程序以正确的顺序查找机场?

有用资源

• python 参考教程 - 该教程包含有关 python 的更多信息:https://www.cainiaomax.com/python/