Python入门教程第44篇 集合推导式

Posted 2022-02-17 不剪发的Tony老师

本篇我们学习如何利用集合推导式（set comprehension）基于已有集合创建一个新的集合。

集合推导式简介

tags 是由三个标签组成的集合：

tags = 'Django', 'Pandas', 'Numpy'

为了将集合中的标签转换为小写形式，可以使用以下 for 循环：

tags = 'Django', 'Pandas', 'Numpy'

lowercase_tags = set()
for tag in tags:
    lowercase_tags.add(tag.lower())

print(lowercase_tags)

输出结果如下：

'django', 'numpy', 'pandas'

以上代码的执行过程如下：

• 首先，遍历集合 tags 中的每个元素。
• 然后，将标签转换为小写形式，然后将其添加到新的集合中。

我们也可以使用内置的 map() 函数加 lambda 表达式实现：

tags = 'Django', 'Pandas', 'Numpy'
lowercase_tags = set(map(lambda tag: tag.lower(), tags))

print(lowercase_tags)

其中，map() 函数返回的是一个映射对象，我们需要使用 set() 函数将其转换为集合。

为了进一步简化代码，Python 提供了集合推导式，语法如下：

expression for element in set if condition

集合推导式可以基于已有集合创建新的集合，它的执行过程如下：

• 首先，遍历集合中的元素。
• 其次，基于每个元素执行表达式。
• 最后，基于表达式的结果创建一个新的元素。

另外，集合推导式支持 if 子句，可以选择满足条件的元素应用表达式并返回相应的结果。

注意，集合推导式返回一个新的集合，而不会修改原始集合。

回到上面的示例，我们可以利用下面的集合推导式将 tags 中的标签转换为小写形式：

tags = 'Django', 'Pandas', 'Numpy'
lowercase_tags = tag.lower() for tag in tags

print(lowercase_tags)

以上语句比 for 循环更加简洁，比 map() 函数更加优雅。

集合推导式与 if 子句

如果我们想要将集合 tags 中除了 Numpy 之外的其他元素转换为小写形式，可以在集合推导式中增加一个 if 条件：

tags = 'Django', 'Pandas', 'Numpy'
new_tags = tag.lower() for tag in tags if tag != 'Numpy'

print(new_tags)

输出结果如下：

'django', 'pandas'

总结

• 集合推导式可以基于已有集合创建一个新的集合。