List Comprehension

List Comprehension

List comprehension offers a shorter syntax when you want to create a new list based on the values of an existing list.

Example:

Based on a list of fruits, you want a new list, containing only the fruits with the letter "a" in the name.

Without list comprehension you will have to write a for statement with a conditional test inside:

With list comprehension you can do all that with only one line of code:

The Syntax

The return value is a new list, leaving the old list unchanged.

The condition is like a filter that only accepts the items that evaluate to True.

The condition if x != "apple"  will return True for all elements other than "apple", making the new list contain all fruits except "apple".

The condition is optional and can be omitted:

The iterable can be any iterable object, like a list, tuple, set etc.

Same example, but with a condition:

The expression is the current item in the iteration, but it is also the outcome, which you can manipulate before it ends up like a list item in the new list:

You can set the outcome to whatever you like:

The expression can also contain conditions, not like a filter, but as a way to manipulate the outcome:

The expression in the example above says:

"Return the item if it is not banana, if it is banana return orange".