推导式是从一个或者多个迭代器快速创建序列的一种方法。它可以将循环和条件判断结合,从而避免冗长的代码。推导式是典型的Python风格,会使用它代表你已经超过Python初学者的水平
列表推导式
列表推导式生成列表对象,语法如下:
[表达式 for item in 可迭代对象]
或者
[表达式 for item in 可迭代对象 if 条件判断]
>>> [x for x in range(1,5)]
[1,2,3,4]
>>> [x*2 for x in range(1,5)]
[2,4,6,8]
>>> [x*2 for x in range(1,20) ifx%5==0]
[10,20,30]
>>> [a for a in "abcdefg"]
['a','b','c','d','e','f','g']
>>> cells=[(row,col) for row in range(1,10) for col in range(1,10)] #可以使用两个循环
>>> for cell in cells:
print(cell)
字典推导式
字典的推导式生成字典对象,格式如下:
{key_expression:value_expression for 表达式 in 可迭代对象}
类似于列表推导式,字典推导也可以增加if条件判断、多个for循环
eg. 统计文本中字符出现的次数:
>>> my_text = "i love you, i love sxt, i love gaoqi"
>>> char_count = {c:my_text.count(c) for c in my_text}
>>> char_count
{' ':9,'i':4,'o':5,'v':3,'e':3,'y':1,'u':1,',':2,'s':1,'x':1,'t':1,'g':1,'a':1,'q':1}
集合推导式
集合推导式生成集合,和列表推导式的语法格式类似:
[表达式 for item in 可迭代对象]
或者
[表达式 for item in 可迭代对象 if 条件判断]
eg.
>>> {x for x in range(1,100) if x%9==0}
{99,36,72,9,45,81,18,54,90,27,63}
生成器推导式(生成元组)
>>> (x for x in range(1,100) if x%9==0)
<generator object <genexpr> at 0x0000000002BD3048>
我们发现提示的是“一个生成器对象”。显然,元组是没有推导式的
一个生成器只能运行一次。第一次迭代可以得到数据,第二次迭代发现数据已经没有了。生成器是可迭代的对象
>>> gnt = (x for x in range(1,100) if x%9==0)
>>> for x in gnt:
print(x, end=' ')
9 18 27 36 45 54 63 72 81 90 99
>>> for x in gnt:
print(x, end=' ')
>>>