欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

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进行数据分析》一书

上一篇:

下一篇: