python从lis中选择特定元素

2024-04-29 07:23:56 发布

您现在位置:Python中文网/ 问答频道 /正文

是否有一种“pythonic”方法可以只从列表中获取某些值,类似于以下perl代码:

my ($one,$four,$ten) = line.split(/,/)[1,4,10]

Tags: 方法代码列表mylinepythoniconeperl
3条回答

使用列表理解

line = '0,1,2,3,4,5,6,7,8,9,10'
lst = line.split(',')
one, four, ten = [lst[i] for i in [1,4,10]]
lst = line.split(',')
one, four, ten = lst[1], lst[4], lst[10]

我想你在找operator.itemgetter

import operator
line=','.join(map(str,range(11)))
print(line)
# 0,1,2,3,4,5,6,7,8,9,10
alist=line.split(',')
print(alist)
# ['0', '1', '2', '3', '4', '5', '6', '7', '8', '9', '10']
one,four,ten=operator.itemgetter(1,4,10)(alist)
print(one,four,ten)
# ('1', '4', '10')

相关问题 更多 >