Python两个列表进行全组合的方法是什么

其他教程   发布日期:2025年02月22日   浏览次数:281

这篇文章主要介绍“Python两个列表进行全组合的方法是什么”,在日常操作中,相信很多人在Python两个列表进行全组合的方法是什么问题上存在疑惑,小编查阅了各式资料,整理出简单好用的操作方法,希望对大家解答”Python两个列表进行全组合的方法是什么”的疑惑有所帮助!接下来,请跟着小编一起来学习吧!

什么是全组合?

两个元组 (a, b)(c, d),则它们的组合有 a,c a,d b,c b,d

方式1

通过itertools类生成生成一个列表列表,其中包含两个列表之间的所有潜在组合

  1. import itertools
  2. list1 = [1, 2, 3, 4]
  3. list2 = [5, 6, 7, 8,9]
  4. print(list(itertools.product(list1, list2)))

输出结果:

  1. [(1, 5), (1, 6), (1, 7), (1, 8), (1, 9), (2, 5), (2, 6), (2, 7), (2, 8), (2, 9), (3, 5), (3, 6), (3, 7), (3, 8), (3, 9), (4, 5), (4, 6), (4, 7), (4, 8), (4, 9)]

方式2

尝试使用列表生成器创建嵌套列表:

  1. >>> [[[x,y] for x in list1] for y in list2]
  2. [[[1, 3], [2, 3]], [[1, 4], [2, 4]]]

或者,如果需要一行列表,只需删除括号:

  1. >>> [[x,y] for x in list1 for y in list2]
  2. [[1, 3], [1, 4], [2, 3], [2, 4]]

方式3

repeat第一个列表,permutate第二个列表和zip它们一起

  1. >>> from itertools import permutations, repeat
  2. >>> a = [1, 2, 3]
  3. >>> b = [4, 5, 6]
  4. >>> list(list(zip(r, p)) for (r, p) in zip(repeat(a), permutations(b)))
  5. [[(1, 4), (2, 5), (3, 6)],
  6. [(1, 4), (2, 6), (3, 5)],
  7. [(1, 5), (2, 4), (3, 6)],
  8. [(1, 5), (2, 6), (3, 4)],
  9. [(1, 6), (2, 4), (3, 5)],
  10. [(1, 6), (2, 5), (3, 4)]]

编辑:正如Peter Otten所指出的,内部zip和repeat是多余的。

  1. [list(zip(a, p)) for p in permutations(b)]

以上就是Python两个列表进行全组合的方法是什么的详细内容,更多关于Python两个列表进行全组合的方法是什么的资料请关注九品源码其它相关文章!