[Python知识库]Python的列表推导式和嵌套列表推导式

推导式
推导式的基本形式：
[expr for val in Collection if condition]
简单翻译一下：
[操作 for 值 in 容器 if 条件]

strings = ['bat','alibaba','as','v','condi']
[x.upper() for x in strings if len(x) > 2]

[]已经说明了是个列表 集合换为{}即可，如下：

str_len = {len(x) for x in strings}
str_len

它等同于利用map函数：set(map(len,strings))

map函数是一种更函数化、更简洁的表达。map（）会根据提供的函数对指定序列做映射。形如map(function, iterable, …)下面是个小例子

def square(x):
    return x ** 2
list(map(square, [1,2,3,4,5]))
list(map(lambda x: x ** 2, [2, 3, 5, 7])) 

一个结合内建序列函数enumerate、将字符串与其位置相匹配的字典作为字典推导式的示例

loc_mapping = {val: index for index, val in enumerate(strings)}
loc_mapping

嵌套列表推导式
嵌套推导式的基本形式：
[expr for collection in collections for val in collection if condition]
再次简单翻译一下：
[操作 for 小容器 in 大容器 for 值 in 小容器 if 条件] 从外到内直到取到最小的单位，有几层就for几次

all_data = [['vince', 'lily','todd', 'josep'],['rik','jason', 'luna', 'steven']]
double_d_name =[name.upper() for names in all_data for name in names if name.count('d') >= 2] 
double_d_name

这是一个包含列表的列表，所以需要for两次，第一次为for 列表 in 总列表，第二次为for name in 列表， 操作为将得到的name变为大写，条件为name中有两个d。

列表推导式是最受欢迎的Python特性之一，但是当嵌套多层之后当考虑代码的可读性。

some_tuple = [(1 ,2, 3), (4, 5, 6), (7, 8, 9)]
[x for tup in some_tuple for x in tup]#拆开元组
[[x for x in tup] for tup in some_tuple]#拆开为多层列表

部分例子引用《利用Python进行数据分析》一书