How do I create a list comprehension in Python?

How do I create a list comprehension in Python?

List comprehensions in Python allow you to create lists in a concise way. For example, to create a list of squares of numbers from 0 to 9, you can use:

squares = [x**2 for x in range(10)].

Can you explain how that works?

Sure! The expression inside the brackets defines the elements of the list. Here, x**2 is the expression that computes the square of each number, and for x in range(10) iterates over numbers from 0 to 9.

What if I want to filter the numbers?

You can add an if condition at the end. For example, to get squares of even numbers only, you can write:

squares_of_evens = [x**2 for x in range(10) if x % 2 == 0].

That's really helpful! Can you give me another example?

Of course! If you want to create a list of the lengths of words in a list, you can do:

word_lengths = [len(word) for word in ['hello', 'world', 'Python']].