百度360必应搜狗淘宝本站头条
当前位置:网站首页 > 技术资源 > 正文

掌握 Python 中的列表推导式

off999 2024-12-04 14:37 18 浏览 0 评论

列表推导式是 Python 中的一项强大功能,它允许在创建和转换列表时使用更清晰、更具可读性的代码。通过将多行代码压缩到单个表达式中,列表推导式使您的代码更加高效和简洁,尤其是在处理数据转换、筛选和映射时。

列表推导式简介

列表推导式是一种创建新列表的简洁方法,它通过将表达式应用于现有列表或可迭代对象的每个元素,并可选择根据条件筛选元素。

语法:

new_list = [expression for item in iterable if condition]
  • expression:应用于每个项目的操作或转换。
  • item:表示 Iterable 中每个元素的变量。
  • iterable:原始集合(如 List、Range 或 Generator)。
  • condition:(可选)根据布尔条件筛选元素。

快速入门:之前和之后的示例

下面是一个比较传统循环与列表推导式的简单示例:

列表推导之前:

numbers = []
for i in range(5):
    numbers.append(i * 2)

列表推导式之后:

numbers = [i * 2 for i in range(5)]

结果:[0, 2, 4, 6, 8]

列表推导式 vs. 传统循环

示例:创建平方数列表

使用传统循环:

squares = []
for x in range(10):
    squares.append(x**2)

使用列表推导式:

squares = [x**2 for x in range(10)]

列表推导式可以提高可读性,并且通常对于简单的任务更有效。但是,对于复杂的 logic,传统 Loop 可能更可取。

性能注意事项和基准测试

由于 Python 的优化,列表推导式通常比传统循环更快。下面是一个使用 timeit 模块的基准测试:

import timeit

sizes = [100, 1000, 10000, 100000]
for size in sizes:
    loop_time = timeit.timeit(f'''
squares = []
for x in range({size}):
    squares.append(x**2)
''', number=100)

    comp_time = timeit.timeit(f'''
squares = [x**2 for x in range({size})]
''', number=100)

    print(f"Size: {size} - Loop: {loop_time:.6f}s, Comprehension: {comp_time:.6f}s")

通常,列表推导式的性能优于传统循环,但差异可能会因复杂性和数据集大小而异。

列表推导式中的高级技术

列表推导式中的嵌套循环

对多维列表或组合元素使用嵌套推导式。

pairs = [(x, y) for x in range(3) for y in range(3)]

具有多个条件的条件逻辑

在单个推导式中应用多个条件来筛选数据。

labels = ["even" if x % 2 == 0 else "odd" for x in range(10)]

内存效率和生成器表达式

生成器表达式是一种节省内存的替代方案,因为它们逐个生成项目,而不是创建整个列表。

# Memory-efficient generator expression
sum(x * 2 for x in range(1000000))

# Memory-intensive list comprehension
sum([x * 2 for x in range(1000000)])

在处理大型数据集时使用生成器表达式以节省内存。

列表推导式的实际应用

数据转型

将摄氏度转换为华氏度:

from typing import List

def process_temperatures(celsius: List[float]) -> List[float]:
    return [(temp * 9/5) + 32 for temp in celsius]

使用 pathlib 进行文件处理

from pathlib import Path

file_sizes = [f.stat().st_size for f in Path('.').glob('*.py')]

使用 defaultdict 对数据进行分组

from collections import defaultdict

names = ['Alice', 'Bob', 'Anna', 'Bill']
by_initial = defaultdict(list)
[by_initial[name[0]].append(name) for name in names]

何时不使用列表推导式

避免对以下对象进行列表推导:

  • 复杂逻辑:对复杂操作使用循环。
  • Multiple Statements(多个语句):推导式只允许一个表达式。
  • 过度嵌套:深度嵌套的推导式会降低可读性。

最佳实践和常见陷阱

最佳实践

  • 保持简单:避免使用复杂的表达式。
  • Limit nesting(限制嵌套):限制嵌套级别。
  • 优先考虑可读性:清晰的条件可以提高可读性。

常见陷阱

可变默认值

 # Correct
[[0] * 3 for _ in range(3)]

# Incorrect
[[0] * 3] * 3  # Same list reference

类型错误:

[int(x) for x in ['1', '2', 'a', '3']]  # ValueError for 'a'

错误处理和故障排除

常见错误和解决方案:

名称错误:确保在使用前定义变量。

[name.title() for name in names]  # NameError if `names` undefined

除法误差

def safe_division(numbers: List[float], divisors: List[float]) -> List[float]:
    return [n/d if d != 0 else float('inf') for n, d in zip(numbers, divisors)]

列表推导式的测试和验证

使用 assert 语句进行验证:

# Testing accuracy
output = [x**2 for x in range(5)]
assert output == [0, 1, 4, 9, 16]

使用 timeit 进行性能分析。

常见模式和习语

数据清洗

def clean_data(raw_data: List[str]) -> List[str]:
    return [entry.strip().lower() for entry in raw_data if entry and not entry.isspace()]

URL 参数解析

def parse_url_params(urls: List[str]) -> List[Dict[str, str]]:
    return [
        dict(param.split('=') for param in url.split('?')[1].split('&'))
        for url in urls 
        if '?' in url
    ]


基本语法:

new_list = [expression for item in iterable if condition]

相关推荐

独家 | 5 个Python高级特性让你在不知不觉中成为Python高手

你已经使用Python编程了一段时间,编写脚本并解决各种问题。是你的水平出色吗?你可能只是在不知不觉中利用了Python的高级特性。从闭包(closure)到上下文管理器(contextmana...

Python装饰器

Python装饰器是一种用于修改函数或类的行为的特殊语法。它们允许在不修改原始代码的情况下,通过将函数或类作为参数传递给另一个函数来添加额外的功能。装饰器本质上是一个函数,它接受一个函数作为参数,并返...

中高阶Python常规用法--上下文管理器

Python以简单性和通用性著称,是一种深受全球开发人员喜爱的编程语言。它提供了大量的特性和功能,使编码成为一种愉快的体验。在这些功能中,一个经常被新手忽视的强大工具是上下文管理器。上下文管理器是高...

Python小案例67- 装饰器

Python装饰器是一种用于修改函数或类的行为的特殊语法。它们允许在不修改原始代码的情况下,通过将函数或类作为参数传递给另一个函数来添加额外的功能。装饰器本质上是一个函数,它接受一个函数作为参数,并返...

python常用的语法糖

概念Python的语法糖(SyntacticSugar)是指那些让代码更简洁、更易读的语法特性,它们本质上并不会增加新功能,但能让开发者更高效地编写代码。推导式写法推导式是Python最经典的...

python - 常用的装饰器 decorator 有哪些?

python编程中使用装饰器(decorator)工具,可以使代码更简洁清晰,提高代码的重用性,还可以为代码维护提供方便。对于python初学者来说,根据装饰器(decorator)的字面意思并不...

python数据缓存怎么搞 ?推荐一个三方包供你参考,非常简单好用。

1.数据缓存说明数据缓存可以说也是项目开发中比不可少的一个工具,像我们测试的系统中,你都会见到像Redis一样的数据缓存库。使用缓存数据库的好处不言而喻,那就是效率高,简单数据直接放在缓存中...

用于时间序列数据的Graphite监视工具

结合第三方工具,Graphite为IT性能监控提供了许多好处。本文介绍其核心组件,包括Carbon、Whisper以及安装的基本准则。Graphite监视工具可实时或按需,大规模地绘制来自多个来源的时...

Python3+pygame实现的坦克大战

一、显示效果二、代码1.说明几乎所有pygame游戏,基本都遵循一定的开发流程,大体如下:初始化pygame创建窗口while循环检测以及处理事件(鼠标点击、按键等)更新UI界面2.代码创建一个m...

Python之鸭子类型:一次搞懂with与上下文装饰器

引言在鸭子类型的理念的基础之上,从关注类型,转变到关注特性和行为。结合Python中的魔法函数的体系,我们可以将自定义的类型,像内置类型一样被使用。今天这篇文章中,接着该话题,继续聊一下with语法块...

Python必会的50个代码操作

学习Python时,掌握一些常用的程序操作非常重要。以下是50个Python必会的程序操作,主要包括基础语法、数据结构、函数和文件操作等。1.HelloWorldprint("Hello,...

一文掌握Python 中的同步和异步

同步代码(Sync)同步就像在一个流水线上工作,每个任务都等待前一个任务完成。示例:机器A切割钢板→完成后,机器B钻孔→完成后,机器C上色。在Python中,同步代码看起来像这样:im...

python 标注模块timeit: 测试函数的运行时间

在Python中,可以使用内置的timeit模块来测试函数的运行时间。timeit模块提供了一个简单的接口来测量小段代码的执行时间。以下是使用timeit测试函数运行时间的一般步骤:导入...

Python带你找回童年的万花尺

还记得小时候的万花尺吧?这么画:一点也不费脑筋,就可以出来这么多丰富多彩的复杂几何图形。具体而言,可以用万花尺玩具(如图2-1所示)来绘制数学曲线。这种玩具由两个不同尺寸的塑料齿轮组成,一大一小。小的...

Python 时间模块深度解析:从基础到高级的全面指南

直接上干货一、时间模块核心类介绍序号类名说明1datetime.datetime表示一个具体的日期和时间,结合了日期和时间的信息。2datetime.date表示一个具体的日期。3datetime.t...

取消回复欢迎 发表评论: