首页 >后端开发 >Python教程 >itertools.product 如何在 Python 中生成多个列表的笛卡尔积?

itertools.product 如何在 Python 中生成多个列表的笛卡尔积?

Susan Sarandon
Susan Sarandon原创
2025-01-04 12:23:38300浏览

How Can itertools.product Generate the Cartesian Product of Multiple Lists in Python?

使用 itertools.product 的多个列表的笛卡尔积

获取一组列表的笛卡尔积,其中每个可能的值组合生成后,利用内置的 itertools.product 函数。此功能自版本 2.6 起已包含在 Python 中。

实现:

import itertools

somelists = [
   [1, 2, 3],
   ['a', 'b'],
   [4, 5]
]

for element in itertools.product(*somelists):
    print(element)

或者,您可以向函数提供单独的参数,如下所示:

for element in itertools.product([1, 2, 3], ['a', 'b'], [4, 5]):
    print(element)

两种方法都会生成相同的结果输出:

(1, 'a', 4)
(1, 'a', 5)
(1, 'b', 4)
(1, 'b', 5)
(2, 'a', 4)
(2, 'a', 5)
...

注意:

  • 此技术通常用于避免深度嵌套循环。
  • 如果您需要笛卡尔同一列表乘以自身的乘积,请考虑使用 itertools.product 并使用解压缩的参数*.
  • 在此上下文中使用 * 来解包参数与其在函数调用中的典型用法没有什么不同。

以上是itertools.product 如何在 Python 中生成多个列表的笛卡尔积?的详细内容。更多信息请关注PHP中文网其他相关文章!

声明:
本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系admin@php.cn