Conditions

Much more complicated lists can be created with an included if statement. The if statement fits right at the end of the statement.

[a for a in range(10) if a % 2 == 0]

## [0, 2, 4, 6, 8]

But what if you need to create an even more complicated list, one that requires an else statement along with the if statement. Then the structure of the list comprehension changes a little, the iterable statement is moved to the end.

[a if a % 2 == 0 else 0 for a in range(10)]

## [0, 0, 2, 0, 4, 0, 6, 0, 8, 0]

Expressions

Of course, expressions can be more complicated than returning single values. One common issue I find is when I have a list of a value type and I need them to be of a different type. This conversion is easy with list comprehensions.

a = ['0', '1', '2', '3', '4']
[int(x) for x in a]

## [0, 1, 2, 3, 4]

There is nearly an unlimited potential of different expressions you can use.

More Iterables

List comprehensions are not limited to a single iterable. Far warning, however, increasing the number of iterables will reduce readability. At some level of complication, it will be a better idea to separate steps.

a = range(5)
b = [5,10,15]
[x*y for x in a for y in b]

## [0, 0, 0, 5, 10, 15, 10, 20, 30, 15, 30, 45, 20, 40, 60]

The results are an element-wise evaluation across multiple iterables. These iterables don’t need to be the same size.

Dictionary Comprehensions

As previously mentioned, by changing the structure, we can generate dictionaries.

{char : num for num, char in enumerate(['a','b','c','d','e'])}

## {'a': 0, 'b': 1, 'c': 2, 'd': 3, 'e': 4}

Likewise, you can create a set rather than a list. Sets can be useful if you don’t need the data to be in order, and you don’t want any duplicate values.

Other Applications

There is great potential in list comprehensions. Often I find that I need to create a list of zeroes or of boolean logic of the same size as a current list. This is easy to create, just don’t refer to the iterable within the expression.

a = range(5)
[True for x in a]

## [True, True, True, True, True]

While it may not be best practice, you can nest a list comprehension within another list comprehension.

[x for x in [b for b in range(20) if b %2 == 0] if x %3 == 0]

## [0, 6, 12, 18]

Conclusions

Hopefully I have won you over with the beauty of list comprehensions. They are simple and clean to create yet extremely flexible in their design. So take a minute, to really appreciate the beauty of list comprehensions.

Photo by Kelly Sikkema on Unsplash