Python List Comprehension

(一)使用List Comprehension的好处　

　在了解Python的List Comprehension之前，我们习惯使用for循环创建列表，比如下面的例子：

numbers = range(10)

my_list = []
for number in numbers:
    my_list.append(number * number)
print(my_list)

可是在Python中，我们有更简洁，可读性更好的方式创建列表，就是List Comprehension：

my_list = [number * number for number in numbers]

我们也可以用map加上lambda实现上述List Comprehension的功能：

my_list = map(lambda a: a*a, numbers)

上面三个代码段的功能类似，除了map函数返回的是iterator，但是从可读性来说，List Comprehension是最好的

 

（二）一些较为复杂的List Comprehension

(1)加上if判断条件的List Comprehension：

my_list = [number for number in numbers if number % 2 == 0]

(2)多个维度，可以包含多个for

colors = ['Black', 'White']
sizes = ['S', 'M', 'L']
tshirts = [(color, size) for color in colors for size in sizes]

 

（三）Dictionary/Set Comprehension

与List Comprehension类似，我们可以对Dictionary和Set操作：

假设我们有如下两个列表：

names = ['Bruce', 'Clark', 'Peter', 'Logan', 'Wade']
heros = ['Batman', 'Superman', 'Spiderman', 'Wolverine', 'Deadpool']

我们利用内置函数zip将上面两个列表打包成包含元组的iterator zip(names, heros) ，用for循环输出如下：

('Bruce', 'Batman')
('Clark', 'Superman')
('Peter', 'Spiderman')
('Logan', 'Wolverine')
('Wade', 'Deadpool')

如果使用for循环把上面的iterator生成字典的话，代码如下：

my_dict = []
for name, hero in zip(names, heros):
    my_dict[name] = hero
print(my_dict)

而用Dictionary Comprehension的等价代码是这样的：

my_dict = {name: hero for name, hero in zip(names, heros)}
print(my_dict)

是不是更为简洁？同样也有Set Comprehension:

numbers = [1, 1, 2, 2, 2, 3, 4, 5]
my_set = {number for number in numbers}
print(my_set)

 

（四）Generator Comprehension

 生成器表达式与列表或其他序列类型相比，更节省内存，因为它一次只产出一个值，与List Comprehension的语法非常类似，只是不用[]，而使用(),

比如我们求字母序列的ASCII码:

letters = 'ABCDEFG'
genexp_1etters = (ord(letter) for letter in letters)
for letter in genexp_1etters:
    print(letter)

程序输出ABCDEF的ASCII码 65 - 71.