python select specific elements from a list

Viewed 116143

Is there a "pythonic" way of getting only certain values from a list, similar to this perl code:

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

I think you are looking for 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')
lst = line.split(',')
one, four, ten = lst[1], lst[4], lst[10]

Try operator.itemgetter (available in python 2.4 or newer):

Return a callable object that fetches item from its operand using the operand’s ____getitem____() method. If multiple items are specified, returns a tuple of lookup values.

>>> 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')

Condensed:

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

Yes:

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

You can also use itemgetter, but I prefer the code above, it's clearer, and clarity == good code.

Using pandas:

import pandas as pd
line = "0,1,2,3,4,5,6,7,8,9,10"
line_series = pd.Series(line.split(','))
one, four, ten = line_series[[1,4,10]]
Related