首页 > 编程语言 >从菜鸟到高手:掌握Python推导式,让代码飞起来,列表、集合、字典,一网打尽,用Python推导式优雅地重构你的数据操作

从菜鸟到高手:掌握Python推导式,让代码飞起来,列表、集合、字典,一网打尽,用Python推导式优雅地重构你的数据操作

时间:2024-08-24 17:23:27浏览次数:14  
标签:推导 Python 菜鸟 元素 列表 numbers 集合 字典

"在Python的广阔世界里,隐藏着一种让程序员们爱不释手的秘密武器——推导式。想象一下,你正站在数据处理的战场上,面对着成千上万条数据,需要快速筛选、转换、聚合。这时,你手中的列表推导、集合推导、字典推导就像三把锋利的剑,轻轻一挥,便能将复杂的数据操作化繁为简,让代码如同行云流水般优雅。今天,就让我们一起揭开这些神奇推导式的面纱,探索它们背后的力量与魅力。"

一、列表推导(List Comprehension

Python中的列表推导(List Comprehension)是一种简洁的构建列表的方法。它允许你从一个已存在的列表(或其他可迭代对象)中快速生成一个新的列表,而不需要编写完整的循环结构。列表推导不仅可以使代码更加简洁,而且在某些情况下还可以提高执行效率。

基本语法:

[expression for item in iterable]
  • expression 是对每一个 item 进行处理的表达式,其结果将被添加到新列表中。
  • item 是可迭代对象 iterable 中的元素。
  • iterable 是一个可以迭代的对象,比如列表、元组、字符串、集合等。

1. 简单的列表推导

假设我们有一个数字列表,想要创建一个新列表,其中包含原列表中每个元素的平方。

使用for循环的实现方法,这样看起来比较繁琐,而且效率比较低

numbers = [1, 2, 3, 4, 5]
numbers2 = []
for num in numbers:
    numbers2.append(num**2)
print(numbers2) # [1, 4, 9, 16, 25]

使用列表推导式,这样的代码看起来就非常的整洁,效率也比较好,而且看起来也会显示有那么一点专业hhh

numbers = [1, 2, 3, 4, 5]
#首先遍历numbers数组,然后将每一个元素赋值给x,x**2,就是numbers数组中每一个元素**2
numbers2 = [x**2 for x in numbers]
print(numbers2) # [1, 4, 9, 16, 25]

2. 带条件的列表推导

可以在列表推导式中加入条件,满足条件的数据才会进入到最终的列表里面

根据numbers列表使用列表推导式生成一个新列表,但是只获取偶数

numbers = [1, 2, 3, 4, 5, 6, 7, 8, 9, 10]
'''
首先先看正常的列表推导式
x for x in numbers
先获取到numbers数组,循环遍历numbers数组,将数组中的每个元素赋值给x
然后x不进行任何操作

接下来就看判断条件
if x % 2 == 0
会判断每一个x,也就是数组中的每一个元素,只有条件为true的元素才会进入到最终的数组中
'''
numbers2 = [x for x in numbers if x % 2 == 0]
print(numbers2) # [2, 4, 6, 8, 10]

3. 嵌套列表推导

列表推导也可以嵌套,以处理更复杂的数据结构。

使用列表推导式获取二维数组中的所有一维数组的元素.,并将每个元素加1然后写入到一个新一维数组中

#定义一个二维数组
numbers = [
    [1,2,3],
    [4,5,6],
    [7,8,9]
]
'''
首先先看 第一个循环
for num in numbers
这段语句就是获取到numbers这个二维数组中的每一个元素,也就是一维数组

然后再看 第二个循环
for x in num
第一个循环把获取到的每一个一维数组赋值给num,现在循环遍历num将每个元素赋值给x

然后x+1,就是每个元素加1,然后赋值给最终的数组
'''
numbers2 = [x+1 for num in numbers for x in num]
print(numbers2) # [2, 3, 4, 5, 6, 7, 8, 9, 10]

在嵌套列表推导中,第一个 for 循环遍历外层的可迭代对象(在这个例子中是 numbers 的每一行),而第二个 for 循环遍历内层的可迭代对象(在这个例子中是每行的元素)。

二、集合推导(Set Comprehension

在Python中,集合推导(Set Comprehension)是一种简洁且强大的工具,用于从一个或多个迭代器快速创建集合(set)。它类似于列表推导(List Comprehension),但结果是一个集合,这意味着结果中的元素是唯一的,且顺序不保证。集合推导的基本语法遵循了集合的特性,即不允许重复元素,并且元素是无序的。

基本语法:

{expression for item in iterable}
  • expression:是一个表达式,用于从每个item中生成新的元素。
  • item:是迭代器iterable中的当前元素。
  • iterable:是一个可迭代对象,如列表、元组、字符串、字典等。
  • condition(可选):是一个条件表达式,用于筛选满足条件的元素。如果条件为真,则当前元素item会被包含在结果集合中。

基本集合推导

# 创建一个包含1到10的平方数的集合  
squared = {x**2 for x in range(1, 11)}  
print(squared)  # 输出可能是 {1, 4, 9, 16, 25, 36, 49, 64, 81, 100},注意顺序可能不同

带条件的集合推导

# 创建一个包含1到10之间偶数的平方的集合  
even_squared = {x**2 for x in range(1, 11) if x % 2 == 0}  
print(even_squared)  # 输出可能是 {4, 16, 36, 64, 100},注意顺序可能不同

从字典推导集合

# 假设我们有一个字典,我们想从它的键中创建一个集合  
d = {'a': 1, 'b': 2, 'c': 3, 'd': 4}  
keys_set = {key for key in d}  
print(keys_set)  # 输出可能是 {'a', 'b', 'c', 'd'},注意顺序可能不同

嵌套集合推导

虽然嵌套集合推导不常见,但它们也是可能的。但请注意,这可能会导致代码难以理解和维护。

# 假设我们有两个列表,我们想找出所有可能的(x, y)对,其中x来自第一个列表,y来自第二个列表  
list1 = [1, 2]  
list2 = [3, 1, 4]  
pairs = {(x, y) for x in list1 for y in list2}  
print(pairs)  # 输出可能是 {(1, 3), (1, 1), (1, 4), (2, 3), (2, 1), (2, 4)},注意顺序可能不同

三、字典推导(Dictionary Comprehension)

Python中的字典推导(Dictionary Comprehension)是一种简洁而强大的方式,用于从可迭代对象(如列表、元组或其他可迭代对象)中创建字典。字典推导类似于列表推导(List Comprehension),但结果是一个字典而不是列表。

语法

{key: value for (key, value) in iterable}

简单的字典推导

假设我们有一个包含元组的列表,每个元组代表一个名字和对应的年龄,我们想将这些数据转换为一个字典:

people = [('Alice', 30), ('Bob', 25), ('Charlie', 35)]  
  
# 使用字典推导创建字典  
people_dict = {name: age for name, age in people}  
  
print(people_dict)  
# 输出: {'Alice': 30, 'Bob': 25, 'Charlie': 35}

包含条件的字典推导

如果我们只想包含年龄大于30岁的人,可以添加条件表达式:

people = [('Alice', 30), ('Bob', 25), ('Charlie', 35)]  
  
# 使用字典推导,并添加条件  
people_over_30 = {name: age for name, age in people if age > 30}  
  
print(people_over_30)  
# 输出: {'Charlie': 35}

复杂的键或值计算

假设我们有一个包含员工ID和姓名的列表,但我们想要将员工ID作为键,并将“Employee_”前缀添加到姓名作为值:

employees = [(1, 'Alice'), (2, 'Bob'), (3, 'Charlie')]  
  
# 使用字典推导,并计算键和值  
employee_dict = {id: 'Employee_' + name for id, name in employees}  
  
print(employee_dict)  
# 输出: {1: 'Employee_Alice', 2: 'Employee_Bob', 3: 'Employee_Charlie'}

嵌套循环和条件

虽然字典推导中直接使用嵌套循环不是直接支持的(因为字典的键必须是唯一的),但你可以通过其他方式(如列表推导或生成器表达式)来间接实现嵌套逻辑,并在字典推导中使用其结果。不过,对于简单的场景,通常建议避免在字典推导中进行过于复杂的嵌套或条件逻辑,以保持代码的可读性。

标签:推导,Python,菜鸟,元素,列表,numbers,集合,字典
From: https://blog.csdn.net/qq_51431069/article/details/141470440

相关文章

  • 【Python数据分析入门】必须掌握的matplotlib绘图语法与常用参数
    掌握绘图基础语法与常用参数掌握pyplot基础语法importmatplotlib.pyplotaspltpyplot基本绘图流程:创建画布与创建子图plt.figure:创建一个空白画布,可指定画布的大小、像素figure.add_sybplot创建并选中子图,可以指定子图的行数、列数和选中图片的编号添加画布内......
  • 基于yolov10的电梯电瓶车、电动车检测系统,支持图像检测,也支持视频和摄像实时检测(pytor
       更多目标检测和图像分类识别项目可看我主页其他文章功能演示:基于yolov10的电梯电瓶车,电动车检测,支持图像、视频和摄像实时检测【pytorch框架、python】_哔哩哔哩_bilibili(一)简介基于yolov10的电梯电瓶车、电动车检测系统是在pytorch框架下实现的,这是一个完整的项目,包......
  • python零基础教学(二)
    元组&列表&字典元组Python的元组与列表类似,不同之处在于元组的元素不能修改,但是元组使用小括号,列表使用方括号,如果你想创建元组,只需要在括号中添加元素,并使用逗号隔开即可元组=(1,2,'哈哈哈')#这就是一个元组,你可以往里面装str,float,int等等列表在元组的基......
  • python常用的算法
    以下是常用的算法及其详细介绍,包括排序算法、查找算法、基础算法和图算法,同时我也会提到每种数据结构的特性、优缺点及使用场景,并给出示例。一、排序算法1.冒泡排序(BubbleSort)冒泡排序是一种简单的排序算法。它通过重复遍历要排序的数列,比较每对相邻元素并交换它们的位......
  • Python爬虫案例二:获取虎牙主播图片(动态网站)
    爬虫流程:优先假设是JSON数据,抓包方式只能翻页JSON数据HTML数据1.异步数据(即先返回HTML,再返回目标的数据,只是触发了JSON请求),不在HTML中2.不能刷新网页,直接翻页测试链接:https://live.huya.com/源代码: importrequests,json,osclassTwo(object):def__ini......
  • 豆瓣评分8.6!Python社区出版的Python故事教程,太强了!
    Python是活力四射的语言,是不断发展中的语言。就连使用Python多年的行者也不敢说对Python的方方面面都了解并可以自由运用,想必读者可能更加无法快速掌握所有重点技巧了。今天给小伙伴们分享的这份手册是用互动的开发故事来探讨Pyfhonic开发的故事书籍,是一本Python语言详解......
  • 豆瓣评分9.0!Python3网络爬虫开发实战,堪称教学典范!
    今天我们所处的时代是信息化时代,是数据驱动的人工智能时代。在人工智能、物联网时代,万物互联和物理世界的全面数字化使得人工智能可以基于这些数据产生优质的决策,从而对人类的生产生活产生巨大价值。在这个以数据驱动为特征的时代,数据是最基础的。数据既可以通过研发产品获得,......
  • GitHub星标破万!Python学习教程(超详细),真的太强了!
    Python是一门初学者友好的编程语言,想要完全掌握它,你不必花上太多的时间和精力。Python的设计哲学之一就是简单易学,体现在两个方面:语法简洁明了:相对Ruby和Perl,它的语法特性不多不少,大多数都很简单直接,不玩儿玄学。切入点很多:Python可以让你可以做很多事情,科学计算和数据......
  • Python爬虫案例一:获取古诗文并按用户输入的作者名进行数据保存
    前言:1、什么是爬虫?也称为网页蜘蛛(WebSpider),通俗来说,解放人的双手,去互联网获取数据,以数据库,txt,excel,csv,pdf,压缩文件,image,video,music保存数据。本质:模拟浏览器,向服务器发送网络请求,接受服务器返回的数据,并保存数据。2、爬虫的分类?A......
  • LeetCode-Python-1650. 二叉树的最近公共祖先 III
    给定一棵二叉树中的两个节点 p 和 q,返回它们的最近公共祖先节点(LCA)。每个节点都包含其父节点的引用(指针)。Node 的定义如下:classNode{publicintval;publicNodeleft;publicNoderight;publicNodeparent;}根据维基百科中对最近公共祖先节点......