列表推导式的用法

在python中，可以使用列表推导式快速简练地创建列表。

例如,我们通常使用下面的方式创建列表：

cities = ['new york','chicago','los angeles','mountain view']
capitalized_cities  = []
for city in cities:
    capitalized_cities.append(city.title())
print(capitalized_cities)

#output
['New York', 'Chicago', 'Los Angeles', 'Mountain View']

使用列表推导式我们可以将创建列表的过程简写为：

capitalized_cities = [city.title() for city in cities]

我们使用方括号 [ ] 创建列表推导式，括号里包含要对可迭代对象中的每个元素进行评估的条件。

列表推导式中的条件语句

我们可以向列表推导式中添加条件语句。在可迭代对象之后，使用关键字 if 检查每次迭代中的条件。

squares = [x**2 for x in range(9) if x % 2 == 0]
print(squares)
#output
[0, 4, 16, 36, 64]

如果你想在上面代码之后直接添加 else，将发生错误。

squares = [x**2 for x in range(9) if x % 2 == 0 else x + 3]

#SyntaxError: invalid syntax

正确添加 else 的方法是将条件语句转移到列表推导式的开头，直接放在表达式后面。

squares = [x**2 if x % 2 == 0 else x + 3 for x in range(9)]
#output
[0, 4, 4, 6, 16, 8, 36, 10, 64]

再举一个例子：

scores = {
             "Rick Sanchez": 70,
             "Morty Smith": 35,
             "Summer Smith": 82,
             "Jerry Smith": 23,
             "Beth Smith": 98
          }
passed = [name for name,score in scores.items() if score >= 65]
print(passed)
#output
['Beth Smith', 'Summer Smith', 'Rick Sanchez']