数据分析师-编程语言与工具-Python_列表推导式与生成器.docx

PAGE1

PAGE1

Python列表推导式基础

1列表推导式的语法结构

列表推导式是Python中一种非常强大的特性，它允许我们以简洁、高效的方式创建列表。列表推导式的语法结构如下：

[expressionforiteminiterableifcondition]

expression：对列表中的每个元素进行的操作，可以是简单的变量引用，也可以是复杂的表达式。

item：从可迭代对象中取出的每个元素。

iterable：可以被迭代的对象，如列表、元组、集合、字典或生成器等。

ifcondition：可选的条件语句，只有当条件为真时，expression的结果才会被添加到列表中。

1.1示例：创建一个平方数列表

假设我们有一个数字列表，我们想要创建一个新的列表，其中包含这些数字的平方。

numbers=[1,2,3,4,5]

squares=[num**2fornuminnumbers]

print(squares)#输出：[1,4,9,16,25]

在这个例子中，expression是num**2，iterable是numbers列表，item是num，没有使用ifcondition。

2列表推导式的应用场景与实例

列表推导式可以用于多种场景，包括数据过滤、数据转换和数据生成等。

2.1示例：数据过滤

假设我们有一个数字列表，我们想要创建一个新的列表，其中只包含偶数。

numbers=[1,2,3,4,5,6,7,8,9,10]

even_numbers=[numfornuminnumbersifnum%2==0]

print(even_numbers)#输出：[2,4,6,8,10]

在这个例子中，expression是num，iterable是numbers列表，item是num，ifcondition是num%2==0。

2.2示例：数据转换

假设我们有一个字符串列表，我们想要创建一个新的列表，其中包含这些字符串的大写形式。

words=[hello,world,python,list,comprehension]

upper_words=[word.upper()forwordinwords]

print(upper_words)#输出：[HELLO,WORLD,PYTHON,LIST,COMPREHENSION]

在这个例子中，expression是word.upper()，iterable是words列表，item是word，没有使用ifcondition。

2.3示例：数据生成

假设我们想要生成一个列表，其中包含1到10之间的所有数字的平方。

squares=[num**2fornuminrange(1,11)]

print(squares)#输出：[1,4,9,16,25,36,49,64,81,100]

在这个例子中，expression是num**2，iterable是range(1,11)，item是num，没有使用ifcondition。

2.4示例：嵌套列表推导式

列表推导式也可以嵌套使用，例如，我们可以创建一个二维列表，其中包含1到3之间的所有数字的平方和立方。

squares_cubes=[[num**2,num**3]fornuminrange(1,4)]

print(squares_cubes)#输出：[[1,1],[4,8],[9,27]]

在这个例子中，expression是[num**2,num**3]，iterable是range(1,4)，item是num，没有使用ifcondition。

2.5示例：列表推导式与字典

列表推导式也可以用于创建字典。例如，我们可以创建一个字典，其中的键是1到5之间的所有数字，值是这些数字的平方。

squares_dict={num:num**2fornuminrange(1,6)}

print(squares_dict)#输出：{1:1,2:4,3:9,4:16,5:25}

在这个例子中，expression是num:num**2，iterable是range(1,6)，item是num，没有使用ifcondition。

列表推导式是一种非常强大且灵活的工具，可以极大地提高代码的可读性和效率。然而，过度使用列表推导式可能会使代码变得难以理解，因此在使用时应保持适度。#Python生成器入门