python中map ()函数的用法
时间: 2023-07-09 07:24:59 浏览: 52
`map()` 函数是 Python 内置的高阶函数,它接收一个函数和一个可迭代对象作为参数,然后返回一个新的可迭代对象,其中每个元素都是对原可迭代对象中相应元素应用函数后的结果。
`map()` 函数的语法如下:
```python
map(function, iterable, ...)
```
其中,`function` 是一个函数,`iterable` 是一个可迭代对象,`...` 表示可以有多个可迭代对象作为参数。
举个例子,假设有一个列表 `nums`,我们想把其中的每个元素都平方一下,可以使用 `map()` 函数实现:
```python
nums = [1, 2, 3, 4, 5]
squares = map(lambda x: x ** 2, nums)
print(list(squares)) # 输出 [1, 4, 9, 16, 25]
```
上面的代码中,`lambda x: x ** 2` 是一个匿名函数,用于对每个元素进行平方运算。`map()` 函数返回的结果是一个迭代器,我们可以通过将其转换为列表来得到最终的结果。
除了使用 lambda 表达式,我们也可以传递一个自定义函数作为 `function` 参数,例如:
```python
def square(x):
return x ** 2
nums = [1, 2, 3, 4, 5]
squares = map(square, nums)
print(list(squares)) # 输出 [1, 4, 9, 16, 25]
```
注意,`map()` 函数返回的是一个迭代器,如果要获取其中的元素,必须使用 `list()`、`tuple()` 或者其他可迭代对象转换函数进行转换。
阅读全文