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

Ответ 1

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

Ответ 2

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

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]]

Ответ 3

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

Ответ 4

Попробуйте 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(','))

Ответ 5

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

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]

Ответ 6

Да:

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

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

Ответ 7

Альтернативно, если у вас был массив 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: Сначала я подумал, что это также будет работать с кортежами, но это немного сложнее. Я предлагаю придерживаться списков или массивов.