如果在python中给出一些元组索引,如何获取子元组列表？

发布时间：2020-12-20 12:31:52 所属栏目：Python 来源：网络整理

导读：有一个元组列表 l = [(1,2,'a','b'),(3,4,'c','d'),(5,6,'e','f')] 我可以用 [(i[0],i[2],i[3]) for i in l] 得到结果 [(1,'f')] 但如果给出一个变量列表,如[0,3],如何得到类似的结果？解决方法像这样使用 operator.itemgetter from operator import itemg

有一个元组列表

l = [(1,2,'a','b'),(3,4,'c','d'),(5,6,'e','f')]

我可以用

[(i[0],i[2],i[3]) for i in l]

得到结果

[(1,'f')]

但如果给出一个变量列表,如[0,3],如何得到类似的结果？

解决方法

像这样使用 operator.itemgetter

>>> from operator import itemgetter
>>> getter = itemgetter(0,3)
>>> [getter(item) for item in l]
[(1,'f')]

如果你有一个索引列表,那么你可以unpack them到itemgetter,就像这样

>>> getter = itemgetter(*[0,3])
>>> [getter(item) for item in l]
[(1,'f')]

（编辑：李大同）

【声明】本站内容均来自网络，其相关言论仅代表作者个人观点，不代表本站立场。若无意侵犯到您的权利，请及时与联系站长删除相关内容!