如何解决Python Itertools 为每个组合添加列表
我需要为 list1
创建每个组合,但将 list2
添加到 list1 的每个组合中。
例如,这会为 list1
的每个值创建一个组合:
list1 = ["a","b","c"]
list2 = ["d","e"]
list(itertools.combinations(list1,2))
[('a','b'),('a','c'),('b','c')]
但我希望结果是:
[('a','b','d','e'),'c'',d','c','e')]
我已经尝试了这些常见的方法,但结果却不尽如人意:
list1 = ["a","e"]
print(list(itertools.combinations(list1,2)).extend(list2))
None
print(list(itertools.combinations(list1,2)) + list2)
[('a','e']
print(list(itertools.combinations(list1,2) + list2))
TypeError: unsupported operand type(s) for +: 'itertools.combinations' and 'list'
任何帮助将不胜感激。谢谢。
解决方法
您可以首先为每个列表分别生成一个大小为 2 的组合的元组,然后计算这些组合列表的笛卡尔积:
import itertools
list1 = ["a","b","c"]
list2 = ["d","e"]
iter1 = itertools.combinations(list1,2)
iter2 = itertools.combinations(list2,2)
# If you want to add a subset of size 2 from list 2:
product = itertools.product(iter1,iter2)
answer = list(map(lambda x: (*x[0],*x[1]),product))
# [('a','b','d','e'),('a','c',('b','e')]
但是,如果您想添加 list2
的所有元素,您可以使用:
import itertools
list1 = ["a","e","f"]
iter1 = itertools.combinations(list1,2)
# If you want to add all elements of list2
product = itertools.product(iter1,[list2])
answer = list(map(lambda x: (*x[0],'e','f'),'f')]
版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 dio@foxmail.com 举报,一经查实,本站将立刻删除。