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的数据置为-1
listb=[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 random
rows,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}