列表生成式
列表生成式即List Comprehensions,是Python内置的非常简单却强大的可以用来创建list的生成式。
举个例子,要生成list [1, 2, 3, 4, 5, 6, 7, 8, 9, 10]可以用range(1, 11):
>>> range(1, 11)
[1, 2, 3, 4, 5, 6, 7, 8, 9, 10]
但如果要生成[1x1, 2x2, 3x3, ..., 10x10]怎么做?方法一是循环:
>>> L = []
>>> for x in range(1, 11):
>>> L.append(x * x)
>>> print(L) #[1, 4, 9, 16, 25, 36, 49, 64, 81, 100]
但是循环太繁琐,而列表生成式则可以用一行语句代替循环生成上面的list:
>>> a=[z*z for z in range(1,10)]
>>> print(a) #[1, 4, 9, 16, 25, 36, 49, 64, 81]
写列表生成式时,把要生成的元素x * x放到前面,后面跟for循环,就可以把list创建出来,十分有用,多写几次,
很快就可以熟悉这种语法。
for循环后面还可以加上if判断,这样我们就可以筛选出仅偶数的平方:
>>> a=[x * x for x in range(1, 11) if x % 2 == 0]
>>> print(a) #[4, 16, 36, 64, 100]
还可以使用两层循环,可以生成全排列:
>>> a=[m + n for m in ‘AB‘ for n in ‘XY‘]
>>> print(a) #[‘AX‘, ‘AY‘, ‘BX‘, ‘BY‘]
***********************
a=(1,2)
b=a
c,d=a
print(b) # (1, 2)
print(c) # 1
print(d) # 2
****************************
for循环其实可以同时使用两个甚至多个变量,比如dict的iteritems()可以同时迭代key和value:
>>> d = {‘x‘: ‘A‘, ‘y‘: ‘B‘, ‘z‘: ‘C‘ }
>>> or k,v in d.items(): #py3中:iteritems变为items
>>> print(k+‘=‘+ v) #x=A y=B z=C
因此,列表生成式也可以使用两个变量来生成list:
>>> d = {‘x‘: ‘A‘, ‘y‘: ‘B‘, ‘z‘: ‘C‘ }
>>> a=[k + ‘=‘ + v for k, v in d.items()]
>>> print(a) #[‘x=A‘, ‘y=B‘, ‘z=C‘]
运用列表生成式,可以快速生成list,可以通过一个list推导出另一个list,而代码却十分简洁。
原文地址:http://blog.51cto.com/13562606/2060261