Python 内建了功能强大的 map() 和 reduce() 函数,它们在数据处理及函数式编程等方面有着重要的应用。
一、map()函数
(一)函数作用及原理
map() 函数接收两个关键参数,一个是函数,另一个是可迭代对象(Iterable),比如列表、元组等。它会将传入的函数依次应用到可迭代对象的每个元素上,然后把这些操作后的结果以新的迭代器(Iterator)形式返回。
(二)示例说明
我们以一个简单的函数 f(x) = x * x 为例,来看看如何将这个函数作用在一个列表 [1, 2, 3, 4, 5, 6, 7, 8, 9] 上。通过以下可视化的示意,能更直观地理解这个过程:
f(x) = x * x
│
│
┌───┬───┬───┬───┼───┬───┬───┬───┐
│ │ │ │ │ │ │ │ │
▼ ▼ ▼ ▼ ▼ ▼ ▼ ▼ ▼
[ 1 2 3 4 5 6 7 8 9 ]
│ │ │ │ │ │ │ │ │
│ │ │ │ │ │ │ │ │
▼ ▼ ▼ ▼ ▼ ▼ ▼ ▼ ▼
[ 1 4 9 16 25 36 49 64 81 ]
用 Python 代码来实现这个过程如下:
def f(x):
return x * x
r = map(f, [1, 2, 3, 4, 5, 6, 7, 8, 9])
print(list(r))
运行上述代码,最终会输出 [1, 4, 9, 16, 25, 36, 49, 64, 81]。这里需要注意的是,map() 函数返回的结果 r 是一个迭代器,迭代器属于惰性序列,它并不会立刻计算并返回所有结果,只有当我们使用 list() 函数去转换它时,才会把整个序列都计算出来,最终以列表的形式呈现。
也许有人会想,不用 map() 函数,通过编写一个循环也能达到同样的计算结果呀,就像下面这样的代码:
L = []
for n in [1, 2, 3, 4, 5, 6, 7, 8, 9]:
L.append(f(n))
print(L)
确实,这样的循环代码也能完成任务。然而,对比一下就能发现,通过 map() 函数实现的代码,能够更加清晰地抽象出运算规则,让人一眼就能明白是在 “把 f(x) 作用在列表的每一个元素,并把结果生成一个新的列表”。而且,map() 函数的优势还在于,它不仅仅局限于计算像 f(x) = x * x 这样简单的函数,还可以轻松处理任意复杂的函数应用场景。
例如,要把一个列表中的所有数字都转为字符串,使用 map() 函数只需一行简洁的代码就能搞定:
print(list(map(str, [1, 2, 3, 4, 5, 6, 7, 8, 9])))
输出结果就是 ['1', '2', '3', '4', '5', '6', '7', '8', '9'],代码简洁明了,充分体现了 map() 函数在处理这类操作时的便捷性。
二、reduce()函数
(一)函数作用及原理
reduce() 函数同样作用于一个序列(如 [x1, x2, x3,...]),不过它需要传入的函数必须能够接收两个参数。reduce() 函数会先将这个函数应用到序列的前两个元素上,然后把得到的结果再与序列的下一个元素继续做累积计算,以此类推,最终将整个序列缩减为一个单一的值。用公式来表示就是 reduce(f, [x1, x2, x3, x4]) = f(f(f(x1, x2), x3), x4)。
(二)示例说明
比如,我们想要对一个序列进行求和操作,就可以借助 reduce() 函数来实现,代码如下:
from functools import reduce
def add(x, y):
return x + y
print(reduce(add, [1, 3, 5, 7, 9]))
运行代码后,会得到总和 25。当然,对于求和运算,Python 本身内置了 sum() 函数,使用起来更加直接简便,在这种情况下可能没必要专门使用 reduce() 函数。
不过,在一些特定的场景下,reduce() 函数就能发挥独特的作用了。例如,要把序列 [1, 3, 5, 7, 9] 变换成整数 13579,就可以利用 reduce() 函数来巧妙地实现,代码如下:
from functools import reduce
def fn(x, y):
return x * 10 + y
print(reduce(fn, [1, 3, 5, 7, 9]))
虽然这个例子看起来实用性不是特别强,但如果我们考虑到字符串 str 其实也可以看作是一个字符序列,对上述例子稍加改动,并配合 map() 函数,就能实现一些很实用的功能,比如写出一个将字符串转换为整数的函数。以下是具体的代码实现过程:
from functools import reduce
DIGITS = {'0': 0, '1': 1, '2': 2, '3': 3, '4': 4, '5': 5, '6': 6, '7': 7, '8': 8, '9': 9}
def char2num(s):
return DIGITS[s]
def str2int(s):
return reduce(lambda x, y: x * 10 + y, map(char2num, s))
print(str2int(['1','5','3','4','5','6','7','8','9']))
153456789
在这个代码中,首先定义了一个字典 DIGITS,用于将字符形式的数字映射为对应的整数。然后通过 char2num 函数实现字符到数字的转换,最后利用 reduce() 函数结合匿名函数(lambda 函数)以及 map() 函数,将字符串中的每个字符依次转换为数字,并按照相应的规则累积计算,最终实现将整个字符串转换为整数的功能。
这意味着,假设 Python 没有提供 int() 函数,我们凭借 map() 和 reduce() 这两个强大的工具,也完全可以自己动手写出一个将字符串转化为整数的函数,而且仅需短短几行代码就能达成,足见它们在编程中的强大作用。
希望通过以上对 Python 中 map() 和 reduce() 函数详细且清晰的介绍,大家能更好地理解并在实际编程中灵活运用这两个函数,发挥它们的优势来处理各种数据处理和转换的任务哦。