I'm trying to figure how to pick all the fruits like:
[['id', 'sub type', 'type'], ['1', 'apples', 'fruit'], ['15', 'orange', 'fruit'], ['3', 'corn', 'vegtable']]
How do I output:
['sub type','apple','orange','corn']
Asked
Active
Viewed 79 times
1
George
- 269
- 2
- 5
- 11
-
1corn isn't a fruit :) – roippi Oct 23 '13 at 00:24
-
1@roippi Yes! This throws an error: `TypeError: Corn is not a fruit`. – aIKid Oct 23 '13 at 00:26
-
1Anyone try 'tomato'...? – smci Oct 23 '13 at 00:28
2 Answers
4
Simple as a list comprehension:
>>> lst = [['id', 'sub type', 'type'], ['1', 'apples', 'fruit'], ['15', 'orange', 'fruit'], ['3', 'corn', 'vegtable']]
>>> [x[1] for x in lst]
['sub type', 'apples', 'orange', 'corn']
>>>
2
operator.itemgetter(1) can be higher-performance
('Trivial answer converted to comment'?! WTF!) This is the one-line reality. Prefer itemgetter to lambda expressions, for performance. Show us more context in which this code occurs, if you can.
