Are list comprehensions syntactic sugar for `list(generator expression)` in Python 3?

by

Both work differently. The list comprehension version takes advantage of the special bytecode LIST_APPEND which calls PyList_Append directly for us. Hence it avoids an attribute lookup to list.append and a function call at the Python level.

>>> def func_lc():
    [x**2 for x in y]
...
>>> dis.dis(func_lc)
  2           0 LOAD_CONST               1 (<code object <listcomp> at 0x10d3c6780, file "<ipython-input-42-ead395105775>", line 2>)
              3 LOAD_CONST               2 ('func_lc.<locals>.<listcomp>')
              6 MAKE_FUNCTION            0
              9 LOAD_GLOBAL              0 (y)
             12 GET_ITER
             13 CALL_FUNCTION            1 (1 positional, 0 keyword pair)
             16 POP_TOP
             17 LOAD_CONST               0 (None)
             20 RETURN_VALUE

>>> lc_object = list(dis.get_instructions(func_lc))[0].argval
>>> lc_object
<code object <listcomp> at 0x10d3c6780, file "<ipython-input-42-ead395105775>", line 2>
>>> dis.dis(lc_object)
  2           0 BUILD_LIST               0
              3 LOAD_FAST                0 (.0)
        >>    6 FOR_ITER                16 (to 25)
              9 STORE_FAST               1 (x)
             12 LOAD_FAST                1 (x)
             15 LOAD_CONST               0 (2)
             18 BINARY_POWER
             19 LIST_APPEND              2
             22 JUMP_ABSOLUTE            6
        >>   25 RETURN_VALUE

On the other hand the list() version simply passes the generator object to list’s __init__ method which then calls its extend method internally. As the object is not a list or tuple, CPython then gets its iterator first and then simply adds the items to the list until the iterator is exhausted:

>>> def func_ge():
    list(x**2 for x in y)
...
>>> dis.dis(func_ge)
  2           0 LOAD_GLOBAL              0 (list)
              3 LOAD_CONST               1 (<code object <genexpr> at 0x10cde6ae0, file "<ipython-input-41-f9a53483f10a>", line 2>)
              6 LOAD_CONST               2 ('func_ge.<locals>.<genexpr>')
              9 MAKE_FUNCTION            0
             12 LOAD_GLOBAL              1 (y)
             15 GET_ITER
             16 CALL_FUNCTION            1 (1 positional, 0 keyword pair)
             19 CALL_FUNCTION            1 (1 positional, 0 keyword pair)
             22 POP_TOP
             23 LOAD_CONST               0 (None)
             26 RETURN_VALUE
>>> ge_object = list(dis.get_instructions(func_ge))[1].argval
>>> ge_object
<code object <genexpr> at 0x10cde6ae0, file "<ipython-input-41-f9a53483f10a>", line 2>
>>> dis.dis(ge_object)
  2           0 LOAD_FAST                0 (.0)
        >>    3 FOR_ITER                15 (to 21)
              6 STORE_FAST               1 (x)
              9 LOAD_FAST                1 (x)
             12 LOAD_CONST               0 (2)
             15 BINARY_POWER
             16 YIELD_VALUE
             17 POP_TOP
             18 JUMP_ABSOLUTE            3
        >>   21 LOAD_CONST               1 (None)
             24 RETURN_VALUE
>>>

Timing comparisons:

>>> %timeit [x**2 for x in range(10**6)]
1 loops, best of 3: 453 ms per loop
>>> %timeit list(x**2 for x in range(10**6))
1 loops, best of 3: 478 ms per loop
>>> %%timeit
out = []
for x in range(10**6):
    out.append(x**2)
...
1 loops, best of 3: 510 ms per loop

Normal loops are slightly slow due to slow attribute lookup. Cache it and time again.

>>> %%timeit
out = [];append=out.append
for x in range(10**6):
    append(x**2)
...
1 loops, best of 3: 467 ms per loop

Apart from the fact that list comprehension don’t leak the variables anymore one more difference is that something like this is not valid anymore:

>>> [x**2 for x in 1, 2, 3] # Python 2
[1, 4, 9]
>>> [x**2 for x in 1, 2, 3] # Python 3
  File "<ipython-input-69-bea9540dd1d6>", line 1
    [x**2 for x in 1, 2, 3]
                    ^
SyntaxError: invalid syntax

>>> [x**2 for x in (1, 2, 3)] # Add parenthesis
[1, 4, 9]
>>> for x in 1, 2, 3: # Python 3: For normal loops it still works
    print(x**2)
...
1
4
9

More Related Contents:

  1. Accessing class variables from a list comprehension in the class definition
  2. how to create a list of lists using list comprehension [closed]
  3. How to create the following lists with list comprehensions? [closed]
  4. Why is “1000000000000000 in range(1000000000000001)” so fast in Python 3?
  5. Are dictionaries ordered in Python 3.6+?
  6. How does asyncio actually work?
  7. The `is` operator behaves unexpectedly with non-cached integers
  8. Why is a list comprehension so much faster than appending to a list?
  9. Two variables in Python have same id, but not lists or tuples
  10. List comprehension vs generator expression’s weird timeit results?
  11. list() uses slightly more memory than list comprehension
  12. yield in list comprehensions and generator expressions
  13. List comprehension and lambdas in Python [duplicate]
  14. Why results of map() and list comprehension are different? [duplicate]
  15. Add an element in each dictionary of a list (list comprehension)
  16. Using while in list comprehension or generator expressions
  17. Why is one class variable not defined in list comprehension but another is?
  18. Is there anything faster than dict()?
  19. Accessing dictionary items by position in Python 3.6+ efficiently
  20. Identifier normalization: Why is the micro sign converted into the Greek letter mu?
  21. Why is a class __dict__ a mappingproxy?
  22. python is operator behaviour with string [duplicate]
  23. Using a function defined in an exec’ed string in Python 3 [duplicate]
  24. How is super() in Python 3 implemented?
  25. list comprehension filtering – “the set() trap”
  26. Appending to a list comprehension in Python returns None [duplicate]
  27. eval fails in list comprehension [duplicate]
  28. List comprehension scope error from Python debugger
  29. Skip elements on a condition based in a list comprehension in python
  30. ‘is’ operator behaves unexpectedly with floats

Leave a Comment