如何解决有没有办法将只有一个元素的列表组合到列表的所有其他元素?
我正在尝试生成列表的第一个元素与所有其他元素的组合。通过首先获得所有可能的组合然后丢弃我不需要的组合,用一个小列表来做这件事不是问题。示例:
import numpy as np
import itertools
lst = ['a','b','c','d','e','f']
element_of_interest = 'a'
all_combinations = np.array(list(itertools.combinations(lst,3)))
idx = np.where(all_combinations == element_of_interest)
combinations_of_interest = all_combinations[idx[0]]
print(combinations_of_interest)
>>> [['a' 'b' 'c']
['a' 'b' 'd']
['a' 'b' 'e']
['a' 'b' 'f']
['a' 'c' 'd']
['a' 'c' 'e']
['a' 'c' 'f']
['a' 'd' 'e']
['a' 'd' 'f']
['a' 'e' 'f']]
但在我的应用程序(290 个元素的列表,r = 4)中,这种做法不够好,因为它非常低效。
谁能想到更好的方法?
解决方法
从列表中取出元素,然后将其添加到所有其余组合中:
In [9]: import itertools
...:
...: lst = ['a','b','c','d','e','f']
...: element_of_interest = 'a'
In [11]: out = [['a'] + list(i) for i in itertools.combinations(lst[1:],2)]
In [12]: out
Out[12]:
[['a','c'],['a','d'],'e'],'f'],'f']]
版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 dio@foxmail.com 举报,一经查实,本站将立刻删除。