Python的列表推导式和嵌套列表推导式
程序员文章站
2024-01-06 13:37:34
...
推导式
推导式的基本形式:
[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进行数据分析》一书