如果您在python中有一个列表,并且想要将索引1、2和5的元素提取到新列表中,您将如何做?
这是我的操作方式,但我不太满意:
>>> a
[10, 11, 12, 13, 14, 15]
>>> [x[1] for x in enumerate(a) if x[0] in [1,2,5]]
[11, 12, 15]
有没有更好的办法?
更一般而言,给定一个索引元组,即使使用重复,您将如何使用该元组从列表中提取相应的元素(例如,元组(1,1,2,1,5)
产生[11,11,12,11,15]
)。
也许使用这个:
[a[i] for i in (1,2,5)]
# [11, 12, 15]
本文收集自互联网,转载请注明来源。
如有侵权,请联系 [email protected] 删除。
我来说两句