首页
学习
活动
专区
圈层
工具
发布
首页
学习
活动
专区
圈层
工具
MCP广场
社区首页 >问答首页 >给定一个列表列表,如何在Python中创建包含所有可能的值组合的矩阵?

给定一个列表列表,如何在Python中创建包含所有可能的值组合的矩阵?
EN

Stack Overflow用户
提问于 2021-07-22 00:50:51
回答 1查看 35关注 0票数 0

我知道如何创建一个组合矩阵,但我想要更有活力的东西。例如,下面创建一个矩阵,其中包含A、B和C的所有可能组合:

代码语言:javascript
复制
A = [0, 1, 2]
B = [3, 5, 7]
C = [10, 20, 30]

MATRIX = []

for a in A:
   for b in B:
      for c in C:
         MATRIX.append([a, b, c])

我想要一些更动态的东西,我可以有A,B,C,.....N列表。我所要定义的就是每个列表中的值,然后我希望它生成一个包含所有可能组合的Matrix。

我该怎么做呢?

EN

回答 1

Stack Overflow用户

发布于 2021-07-22 00:54:23

使用itertools.product,您可以传递任意数量的迭代器:

代码语言:javascript
复制
from itertools import product
print(list(product(A, B, C)))

输出:

代码语言:javascript
复制
[(0, 3, 10), (0, 3, 20), (0, 3, 30), (0, 5, 10), (0, 5, 20), (0, 5, 30), (0, 7, 10), (0, 7, 20), (0, 7, 30), (1, 3, 10), (1, 3, 20), (1, 3, 30), (1, 5, 10), (1, 5, 20), (1, 5, 30), (1, 7, 10), (1, 7, 20), (1, 7, 30), (2, 3, 10), (2, 3, 20), (2, 3, 30), (2, 5, 10), (2, 5, 20), (2, 5, 30), (2, 7, 10), (2, 7, 20), (2, 7, 30)]

如果你想要一个列表列表,而不是元组列表,只要写一个简单的“列表理解”即可。

票数 2
EN
页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持
原文链接:

https://stackoverflow.com/questions/68473538

复制
相关文章

相似问题

领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档