145

Consider the following Python code with which I add in a new list2 all the items with indices from 1 to 3 of list1:

for ind, obj in enumerate(list1):
    if 4 > ind > 0:
        list2.append(obj)

How would you write this using list comprehension, if I have no access to the indices through enumerate?

something like:

list2 = [x for x in list1 if 4 > ind > 0]

but since I have no ind number, would this work?

list2 = [x for x in enumerate(list1) if 4 > ind > 0]
Rodrigue
  • 3,617
  • 2
  • 37
  • 49
Pav Ametvic
  • 1,727
  • 4
  • 12
  • 14

4 Answers4

261
list2 = [x for ind, x in enumerate(list1) if 4 > ind > 0]
Pavel Anossov
  • 60,842
  • 14
  • 151
  • 124
48

If you use enumerate, you do have access to the index:

list2 = [x for ind, x in enumerate(list1) if 4>ind>0]
BrenBarn
  • 242,874
  • 37
  • 412
  • 384
14

Unless your real use case is more complicated, you should just use a list slice as suggested by @wim

>>> list1 = ['zero', 'one', 'two', 'three', 'four', 'five', 'six']
>>> [x for ind, x in enumerate(list1) if 4 > ind > 0]
['one', 'two', 'three']
>>> list1[1:4]
['one', 'two', 'three']

For more complicated cases - if you don't actually need the index - it's clearer to iterate over a slice or an islice

list2 = [x*2 for x in list1[1:4]]

or

from itertools import islice
list2 = [x*2 for x in islice(list1, 1, 4)]

For small slices, the simple list1[1:4]. If the slices can get quite large it may be better to use an islice to avoid copying the memory

John La Rooy
  • 295,403
  • 53
  • 369
  • 502
  • thanks but since I want to perform an operation on 'x' inside the comprehension (lets say x*x) before I store it inside the new list, it looks like using the slice inside the list comprehension is the best way ! thanks – Pav Ametvic Feb 14 '13 at 00:19
  • @PavAmetvic, ok those cases are still simple enough to not need enumeration – John La Rooy Feb 14 '13 at 00:36
6

For those wondering, you can just as easily store the matching indices as the result of the list comprehension if your use-case requires it. It may be that you need to evaluate the value of each item in the list to determine if it matches some value or pattern, and that you just want to obtain a list of the matching indices for later processing rather than the matching values.

This can be done as shown below:

>>> values = ['zero', 'one', 'two', 'three', 'four', 'five', 'six']
>>> search = ['one', 'three', 'five']
>>> matches = [i for i, x in enumerate(values) if x in search]
>>> print(matches)
[1, 3, 5]

The important part above is the use of the [i for i, x in enumerate(l) if ...] pattern in the list comprehension, where the index i is being captured rather than the value x which would be achieved using the more typical [x for x in l if ...] or the alternate [x for i, x in enumerate(l) if ...] enumeration list comprehension patterns.

bluebinary
  • 1,026
  • 14
  • 17