Data Structures

List Comprehension

A list comprehension builds a new list in one readable line.

Syntax:

[expression for item in iterable if condition]

numbers = [1, 2, 3, 4, 5, 6]

squares = [n * n for n in numbers]
evens = [n for n in numbers if n % 2 == 0]
names = [name.title() for name in ["ali", "sara"]]

print(squares)
print(evens)
print(names)

Output

[1, 4, 9, 16, 25, 36]
[2, 4, 6]
['Ali', 'Sara']