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

本篇我们学习如何利用集合推导式(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'}

总结

  • 集合推导式可以基于已有集合创建一个新的集合。
上一篇:classpath属性的作用


下一篇:Beautiful Soup 库