本篇我们学习如何利用集合推导式(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'}
总结
- 集合推导式可以基于已有集合创建一个新的集合。