python学习笔记--推导式(Comprehension)

列表推导式

创建列表或者从已知列表中筛选数据，通常最简单的方法是使用列表推导式。

例如创建一个自然数的平方的列表，可以这样写：

lista=[n**2 for n in range(21)]print(lista) # [0, 1, 4, 9, 16, 25, 36, 49, 64, 81, 100, 121, 144, 169, 196, 225, 256, 289, 324, 361, 400] 

从列表中筛选数据：

# 筛选小于等于100的数据且将大于100的数据置为-1listb=[n if n <=100 else -1 for n in lista]print(listb)# [0, 1, 4, 9, 16, 25, 36, 49, 64, 81, 100, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1]

同样，在生成器中也可以使用推导式，例如：

listc=(n for n in listb if n<0)print(listc) # <generator object <genexpr> at 0x02C2FD20>for n in listc:    print(n,end=' ')# -1 -1 -1 -1 -1 -1 -1 -1 -1 -1 

注意：当推导式中只有if条件限制时，if语句应该放到for的后面。当有if..else限制时，应该放到for语句的前面。

当筛选条件比较复杂时，可将其放到函数中处理，例如从列表中筛选整数：

nums=[1,2,3,4,5,'6','7','TEST','-9','123.123']listd=[n for n in nums if isInt(n)]print(listd)# [1, 2, 3, 4, 5, '6', '7', '-9']

或者使用内建函数filter()，例如：

liste=list(filter(isInt,nums))print(liste)# [1, 2, 3, 4, 5, '6', '7', '-9']

使用列表推导式实现二维数组的转置，例如：

import randomrows,columns=3,4 # 3行4列array2d=[[random.randrange(10) for col in range(columns)] for row in range(rows)]print(array2d)# [[7, 7, 2, 7], [3, 0, 6, 7], [9, 2, 4, 1]]aT=[[row[i] for row in array2d] for i in range(len(array2d[0]))]print(aT)# [[7, 3, 9], [7, 0, 2], [2, 6, 4], [7, 7, 1]]

字典推导式：

当从一个已知字典筛选数据并创建一个新的字典时，可以使用字典推导式，例如：

data={    'hilary':19,'vergil':24,'mike':12,'tom':6,'jerry':7}data2={key:val for key,val in data.items() if val>10}print(data2) # {'hilary': 19, 'vergil': 24, 'mike': 12}