博客

当我们需要将一个 Python List 中的每个元素进行处理得到一个新的 List 时，可以使用 map 函数。下面是 map 函数的基本用法示例：

map(function, iterable)

其中，function 表示一个函数，iterable 表示一个可迭代对象，如 List，Tuple 等。map 函数逐一对 iterable 中的每个元素应用 function 函数，并返回一个新的 List。

以下是两个示例说明：

1. 将 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 类型返回。

1. 将 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 类型返回。