python выбирает определенные элементы из списка

Я думаю, что вы ищете 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') 

Использование понимания списка

 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 (доступно в python 2.4 или новее):

Возвращает вызываемый объект, который извлекает элемент из его операнда, используя метод ____ getitem ____ () операнда. Если указано несколько элементов, возвращает кортеж значений поиска.

 >>> from operator import itemgetter >>> line = ','.join(map(str, range(11))) >>> line '0,1,2,3,4,5,6,7,8,9,10' >>> a, b, c = itemgetter(1, 4, 10)(line.split(',')) >>> a, b, c ('1', '4', '10') 

Сгущенное:

 >>> # my ($one,$four,$ten) = line.split(/,/)[1,4,10] >>> from operator import itemgetter >>> (one, four, ten) = itemgetter(1, 4, 10)(line.split(',')) 

Как насчет этого:

 index = [1, 0, 0, 1, 0] x = [1, 2, 3, 4, 5] [i for j, i in enumerate(x) if index[j] == 1] #[1, 4] 

Да:

 data = line.split(',') one, four, ten = data[1], data[4], data[10] 

Вы также можете использовать itemgetter, но я предпочитаю код выше, яснее и ясности == хороший код.

Кроме того, если у вас был массив Numpy вместо списка, вы могли бы сделать что-то вроде:

 from numpy import array # Assuming line = "0,1,2,3,4,5,6,7,8,9,10" line_array = array(line.split(",")) one, four, ten = line_array[[1,4,10]] 

Фокус в том, что вы можете передать список (или массив Numpy) в виде индексов массива.

EDIT: Сначала я подумал, что это также будет работать с кортежами, но это немного сложнее. Я предлагаю придерживаться списков или массивов.