python--推导式

推导式

推导式是从一个或者多个迭代器快速简洁地创建数据结构的一种方法。它可以

将循环和条件判断结合，从而避免语法冗长的代码。

可以结合range() 函数使用一个迭代器：

>>> number_list = []
>>> for number in range(1, 6):
... number_list.append(number)
...
>>> number_list
[1, 2, 3, 4, 5]
num_list = list(range(0,10))
print(num_list)

上面这些方法都是可行的Python 代码，会得到相同的结果。然而，更像Python 风格的创

建列表方式是使用列表推导。最简单的形式如下所示：

[ expression for item in iterable ]
>>> number_list = [number for number in range(1,6)]
>>> number_list
[1, 2, 3, 4, 5]
########################
>>> number_list = [number-1 for number in range(1,6)]
>>> number_list
[0, 1, 2, 3, 4]

列表推导也可以像下面的例子加上条件表达式：

[expression for item in iterable if condition]
>>> a_list = [a for a in range(0,11) if a%2 ==0]
>>> a_list
[0, 2, 4, 6, 8, 10]
>>> a_list = [a-10 for a in range(1,11) if a%2 ==0]
>>> a_list
[-8, -6, -4, -2, 0]
####################################
rows = range(1,4)
cols = range(1,3)
>>> for row in rows:
... for col in cols:
... print(row,col)
...
1 1
1 2
2 1
2 2
3 1
3 2
#########################################
>>> cells =[(row,col) for row in rows for col in cols]
>>> print(cells)
[(1, 1), (1, 2), (2, 1), (2, 2), (3, 1), (3, 2)]
>>> for cell in cells:
... print(cell)
...
(1, 1)
(1, 2)
(2, 1)
(2, 2)
(3, 1)
(3, 2)

字典推导式

除了列表，字典也有自己的推导式。最简单的例子就像：

{ key_expression : value_expression for expression in iterable }
类似于列表推导，字典推导也有if 条件判断以及多个for 循环迭代语句：
>>> word = 'letters'
>>> letter_counts = {letter: word.count(letter) for letter in word}
>>> letter_counts
{'l': 1, 'e': 2, 't': 2, 'r': 1, 's': 1}

集合推导式

集合也不例外，同样有推导式。最简单的版本和之前的列表、字典推导类似：

{expression for expression in iterable }
最长的版本（if tests, multiple for clauses）对于集合而言也是可行的：
>>> a_set = {number for number in range(1,6) if number % 3 == 1}
>>> a_set
{1, 4}

生成器推导式

元组是没有推导式的。你可能认为将列表推导式中的方括号变成圆括号就可以定义

元组推导式，就像下面的表达式一样：

>>> number_thing = (number for number in range(1, 6))
其实，圆括号之间的是生成器推导式，它返回的是一个生成器对象：
>>> type(number_thing)
<class 'generotor’>

练习

(1) 将7 赋值给变量guess_me，然后写一段条件判断（if、else 和elif）的代码：如果guess_me 小于7 输出’too low’；大于7 则输出’too high’；等于7 则输出’just right’。

(2) 将7 赋值给变量guess_me，再将1 赋值给变量start。写一段while 循环代码，比较start和guess_me：如果start 小于guess_me，输出too low； 如果等于则输出’found it!’ 并终止循环；如果大于则输出’oops’，然后终止循环。在每次循环结束时自增start。

(3) 使用 for 循环输出列表[3, 2, 1, 0] 的值。

(4) 使用列表推导生成0~9（range(10)）的偶数列表。

(5) 使用集合推导创建集合odd，包含0~9（range(10)）的奇数。

(6) 使用字典推导创建字典squares，把0~9（range(10)）的整数作为键，每个键的平方作为对应的值。