推导式
推导式是从一个或者多个迭代器快速简洁地创建数据结构的一种方法。它可以
将循环和条件判断结合,从而避免语法冗长的代码。
可以结合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’>
网友评论