如何解决Python itertools.combinations:如何同时获取组合内组合数的索引
根据此处提出的问题:Python itertools.combinations: how to obtain the indices of the combined numbers,给出以下代码:
import itertools
my_list = [7,5,4]
pairs = list(itertools.combinations(my_list,2))
#pairs = [(7,5),(7,4),(5,4)]
indexes = list(itertools.combinations(enumerate(my_list ),2)
#indexes = [(0,1),(0,2),3),(1,(2,3)]
有什么方法可以在一行中获得 pairs
和 indexes
,这样我的代码的复杂度就可以降低(例如使用 enumerate
或类似的东西)?
解决方法
@Maf - 试试这个,正如@jonsharpe 之前建议的那样,使用 zip
:
from pprint import pprint
from itertools import combinations
my_list = [7,5,4]
>>> pprint(list(zip(combinations(enumerate(my_list),2),combinations(my_list,2))))
[(((0,7),(1,5)),(7,(((0,(2,(3,4)),(((1,5),(5,(((2,4))]
,
(显式优于隐式。简单优于复杂。) 我会使用列表理解的灵活性:
list((x,(x[0][1],x[1][1])) for x in list(combinations(enumerate(my_list),2)))
这可以使用 opertor.itemgetter
之类的东西进一步扩展。
此外,这个想法是只运行一次使用迭代器,以便该方法也可以应用于其他非确定性迭代器,例如,来自 yield
的 random.choices
。>
版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 dio@foxmail.com 举报,一经查实,本站将立刻删除。