When you need to do something involving a list in Python, either process an existing list or create a new list, there’s a kind of shorthand that can be used to do this in a single line between square brackets, called ‘List Comprehension’. Think of it as For loops within list brackets, with or without conditionals.
# given a list casual_names = ['alec','jude','malcolm'] capitalized_names = [name.title() for name in casual_names]
You can add a conditional statement. For example, square all the even integers in a given range:
squares = [x**2 for x in range(16) if x % 2 == 0]
You can use if else statements, but they have to appear before the for. Convert positive and negative reviews to integers:
# for a list of reviews reviews = ['positive','negative','positive'] # encode them as 1's and 0's encoded_reviews = [1 if r == "positive" else 0 for r in reviews]