python中推导式的用法
推导式创建序列
推导式是从一个或者多个迭代器快速创建序列的一种方法。它可以将循环和条件判断结合,从而避免冗长的代码。推导式是典型的 Python 风格
列表推导式
列表推导式生成列表对象,语法如下:
[表达式foritemin 可迭代对象 ]或者:{表达式foritemin 可迭代对象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) if x%5==0 ]
[10, 20, 30]
>>> [aforain "abcdefg"]
['a', 'b', 'c', 'd', 'e', 'f', 'g']
>>> cells = [(row,col) forrow in range(1,10) for col in range(1,10)]#可以使用两个循环
>>> for cell in cells:
print(cell)
字典推导式
字典的推导式生成字典对象,格式如下:
{key_expression:value_expressionfor表达式in可迭代对象}
类似于列表推导式,字典推导也可以增加 if 条件判断、多个 for 循环。
统计文本中字符出现的次数:
>>> my_text = ' i love you, i love china, i love gao'
>>> char_count = {c:my_text.count(c)for c in my_text}
>>> char_count
{' ': 9, 'i': 4, 'l': 3, 'o': 5, 'v': 3, 'e': 3, 'y': 1, 'u': 1, ',': 2, 'h': 1, 'n': 1, 'c': 1, 'g': 1, 'a': 2}
集合推导式
集合推导式生成集合,和列表推导式的语法格式类似:
>>> {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=' ')