reduce
函数原本在python2
中也是个内置函数,不过在python3
中被移到functools
模块中。
reduce
函数先从列表(或序列)中取出2
个元素执行指定函数,并将输出结果与第3
个元素传入函数,输出结果再与第4
个元素传入函数,...,以此类推,直到列表每个元素都取完。
对列表元素求和,如果不用reduce
,我们一般常用的方法是for
循环:
def sum_func(arr):
if len(arr) <= 0:
return 0
else:
out = arr[0]
for v in arr[1:]:
out += v
return out
a = [1, 2, 3, 4, 5]
print(sum_func(a))
可以看到,代码量比较多,不够优雅。如果使用reduce
,那么代码将非常简洁:
from functools import reduce
a = [1, 2, 3, 4, 5]
def add(x, y): return x + y
print(reduce(add, a))
输出结果为:
15
与内置函数map
和filter
不一样的是,在性能方面,reduce
相比较for
循环来说没有优势,甚至在实际测试中
reduce
比for
循环更慢。
from functools import reduce
import time
def test_for(arr):
if len(arr) <= 0:
return 0
out = arr[0]
for i in arr[1:]:
out += i
return out
def test_reduce(arr):
out = reduce(lambda x, y: x + y, arr)
return out
a = [i for i in range(100000)]
t1 = time.perf_counter()
test_for(a)
t2 = time.perf_counter()
test_reduce(a)
t3 = time.perf_counter()
print('for循环耗时:', (t2 - t1))
print('reduce耗时:', (t3 - t2))
输出结果如下:
for循环耗时: 0.009323899999999996
reduce耗时: 0.018477400000000005
因此,如果对性能要求苛刻,建议不用reduce
, 如果希望代码更优雅而不在意耗时,可以用reduce
。
如果您觉得本文对你有帮助,欢迎关注我【Python学习实战】,第一时间获取最新更新。每天学习一点点,每天进步一点点。
原创声明:本文系作者授权腾讯云开发者社区发表,未经许可,不得转载。
如有侵权,请联系 cloudcommunity@tencent.com 删除。
原创声明:本文系作者授权腾讯云开发者社区发表,未经许可,不得转载。
如有侵权,请联系 cloudcommunity@tencent.com 删除。