python列表推导式、集合推导式和字典推导式

版权声明:本文为博主原创文章,需要转载麻烦先评论留言告知哦~ https://blog.csdn.net/qq_37315403/article/details/81592378

列表推导式(List comprehensions)

通过列表推导式我们能更方便简洁地创建列表,一般是对序列(字符串、列表、元组)或者是可迭代对象(py3中range()返回就是可迭代对象)中的元素进行某些操作来生成新的列表。

假如我们想快速创建一个包含0到5的平方值的列表,我们可以:

squares = []
for x in range(6):    #循环添加元素
    squares.append(x**2)
print(squares)      #[0, 1, 4, 9, 16, 25]

或者我们可以选择更简洁也更易读的方式:

squares = [x**2 for x in range(6)]  #直接使用列表推导式
print(squares)      #[0, 1, 4, 9, 16, 25]

当然,通过列表推导式我们还可以有更多操作:

vec = [-4, -2, 0, 2, 4]  #创建一个列表

#筛选出vec列表中所有大于0的元素
print([x for x in vec if x > 0])   #[2, 4]

#对vec列表每个元素应用函数,此处为求绝对值
print([abs(x) for x in vec])   #[4, 2, 0, 2, 4]

#为列表每个元素调用方法,此处为转换字母小写
freshfruit = ['baNana', 'loGANberRY ', 'pAssiON fRUIT']
print([fruit.lower() for fruit in freshfruit])   
#输出:['banana', 'loganberry ', 'passion fruit']

#生成两个列表中不相等元素可以组成的所有对的列表
print([(x, y) for x in [1,2,3] for y in [3,1,4] if x != y])   
#输出:[(1, 3), (1, 4), (2, 3), (2, 1), (2, 4), (3, 1), (3, 4)]

此外,我们还可以通过嵌套列表推导式来进行一些操作:

#构造一个二维列表模拟二维数组
matrix = [[1, 2, 3, 4],
          [5, 6, 7, 8],
          [9, 10, 11, 12]]
#通过嵌套列表推导式来进行
print([ [row[i] for row in matrix] for i in range(len(matrix[0])) ])
#[[1, 5, 9], [2, 6, 10], [3, 7, 11], [4, 8, 12]]

总结:使用列表推导其实很简单,记住框架然后灵活使用即可。值得注意的是每个部分之间的的空格必不能少。
个人归纳的语法框架(仅供参考):[对单个元素的操作 for 单个元素 in 原始序列 if 筛选条件]


集合推导式

集合推导式和列表推导式极其相似,用法也完全一样,唯一的区别就在于将代表列表的符号[ ]替换成代表几个的符号{ }

vec = [-4, -2, 0, 2, 4]  #创建一个列表

#筛选出vec列表中所有大于0的元素,结果为列表
print([x for x in vec if x > 0])   #[2, 4]
#筛选出vec列表中所有大于0的元素,结果为集合
print({x for x in vec if x > 0})   #{2, 4}

#对vec列表每个元素求绝对值,结果为列表
print([abs(x) for x in vec])   #[4, 2, 0, 2, 4]
#对vec列表每个元素求绝对值,结果为集合
print({abs(x) for x in vec})   #{0, 2, 4}  注意此处的集合自动去重

总结:要注意的只有一点,使用集合推导式的返回结果会自动去重。


字典推导式

通过上面的介绍,大概可以猜出来字典推导式该怎么用了,直接来试试吧:

#生成以x为键,x平方为值的字典
print({x: x**2 for x in (2, 4, 6)})
#输出:{2: 4, 4: 16, 6: 36}

#快速将两个关联的列表以键值对的形式生成字典
names = ['trophy','jack','bob']
ages = [19,21,23]
print({name:age for (name, age) in zip(names,ages)})
#输出:{'trophy': 19, 'jack': 21, 'bob': 23}

参考:python官方文档(英文)

阅读更多
想对作者说点什么?

博主推荐

换一批

没有更多推荐了,返回首页