当我们需要将一个 Python List 中的每个元素进行处理得到一个新的 List 时,可以使用 map
函数。下面是 map
函数的基本用法示例:
map(function, iterable)
其中,function
表示一个函数,iterable
表示一个可迭代对象,如 List,Tuple 等。map
函数逐一对 iterable
中的每个元素应用 function
函数,并返回一个新的 List。
以下是两个示例说明:
- 将 List 中的元素求平方
lst = [1, 2, 3, 4, 5]
res = list(map(lambda x: x * x, lst))
print(res) # [1, 4, 9, 16, 25]
上述代码中,使用 map
函数对 lst
中的每个元素执行平方操作,使用了 lambda 匿名函数来实现。最后使用 list
函数将 map 返回的结果转换成 List 类型返回。
- 将 List 中的所有元素转换成字符串,并添加前缀和后缀
lst = [1, 2, 3, 4, 5]
res = list(map(lambda x: "val_" + str(x) + "_end", lst))
print(res) # ['val_1_end', 'val_2_end', 'val_3_end', 'val_4_end', 'val_5_end']
上述代码中,使用 map
函数对 lst
中的每个元素执行转换和拼接操作,使用了 lambda 匿名函数来实现。最后使用 list
函数将 map 返回的结果转换成 List 类型返回。