python中的map函数

# 使用map函数将列表中的每个元素平方

# 定义一个函数，用于计算平方
def square(x):
    return x ** 2

# 原始列表
numbers = [1, 2, 3, 4, 5]

# 使用map函数将列表中的每个元素传递给square函数，并返回一个新的迭代器
squared_numbers_iterator = map(square, numbers)

# 将迭代器转换为列表以便查看结果
squared_numbers = list(squared_numbers_iterator)

print(squared_numbers)  # 输出: [1, 4, 9, 16, 25]

# 也可以使用lambda表达式来简化代码
squared_numbers_lambda = list(map(lambda x: x ** 2, numbers))

print(squared_numbers_lambda)  # 输出: [1, 4, 9, 16, 25]

解释说明：

• map(function, iterable) 函数会将 iterable 中的每个元素传递给 function，并返回一个新的迭代器，其中包含应用函数后的结果。
• 在上面的例子中，我们首先定义了一个 square 函数，用于计算平方。然后我们使用 map 函数将 numbers 列表中的每个元素传递给 square 函数，并将结果存储在 squared_numbers_iterator 中。
• 最后，我们将 squared_numbers_iterator 转换为列表以便查看结果。
• 我们还展示了如何使用 lambda 表达式来简化代码，避免定义额外的函数。