Python map()应用于集合

  • Post category:Python

Python中的map()函数可以将一个函数应用到一个集合中的每个元素,然后将得到的结果作为一个新的集合返回。

下面是map()函数的基本语法:

map(function, iterable, ...)

其中,第一个参数function是一个函数,表示要应用到集合中每个元素上的操作;第二个参数iterable则是一个可迭代的对象,比如列表、元组、集合等。在函数应用之后,map()函数将返回一个迭代器对象,可以通过list()函数将其转换为列表。

以下是两个map()函数的示例说明:

  1. 把整数列表中的每个元素平方
lst = [1, 2, 3, 4, 5]
squared = list(map(lambda x: x ** 2, lst))
print(squared)
# 输出:[1, 4, 9, 16, 25]

上面的代码中,我们使用map()函数将一个匿名函数应用到整数列表lst中的每个元素上,并将结果存储在squared列表中。该匿名函数使用lambda关键字定义,其功能是将输入参数的平方返回。

  1. 把字符串列表中的每个元素转换为大写
words = ["apple", "banana", "cherry"]
uppercase = list(map(str.upper, words))
print(uppercase)
# 输出:['APPLE', 'BANANA', 'CHERRY']

此处,我们使用map()函数将str类中的内置upper()函数应用到列表words中的每个字符串元素上,并将结果存储在uppercase列表中。这个upper()函数的作用是将字符串的所有字符转换为大写字母。

总结起来,map()函数是Python中一个十分实用的函数,常常和匿名函数或内置函数结合使用。使用它可以有效地简化代码,使代码更加优雅。