Python编程技巧:如何用Map, Filter, Reduce代替For循环?-中国科技新闻网

你是否有过这样的经历,你查看自己写的代码并看到满眼的 for 循环?你发现你必须斜着你的眼睛,并将脑袋前倾到你的显示器,以看得更清楚。

反正我有过这样的经历。

for 循环就像是一把瑞士军刀,它可以解决很多问题,但是,当你需要扫视代码,快速搞清楚代码所做的事情时,它们可能会让人不知所措。

map、filter 和 reduce 这三种技术可以提供描述迭代原因的函数替代方案,以便避免过多的 for 循环。我之前在 JavaScript 中写过这些技术的入门文章,但是它们在 Python 中的实现略有不同。

我们将简要介绍这三种技术,主要介绍它们在 JavaScript 和 Python 中的语法差异,然后给出如何转换 for 循环的示例。

什么是 Map、Filter 和 Reduce?

回顾我以前编写的代码,我意识到 95% 的时间都花在遍历字符串或数组上。在这种情况下,我会执行以下操作之一:将一系列语句映射到每个值,筛选满足特定条件的值,或将数据集减少为单个聚合值。

有了这种洞察力,你就可以识别和实现这三种方法,即循环遍历通常属于这三种功能类别之一:

  • Map:对每个项应用相同的步骤集,存储结果

  • Filter:应用验证条件,存储计算结果为 True 的项

  • Reduce:返回一个从元素传递到元素的值

为什么 Python Map/Filter/Reduce 会不一样?

在 Python 中,这三种技术作为函数存在,而不是数组或字符串类的方法。这意味着,你将编写 map(function, my_list),而不是编写 my_array.map(function)。

此外,每个技术都需要传递一个函数,该函数将执行每个项目。通常,该函数是作为匿名函数(在 JavaScript 中称为 arrow 头函数)编写的。但是,在 Python 中,你经常看到被使用的是 lambda 表达式。

lambda 表达式和 arrow 函数之间的语法实际上非常相似。将 => 替换为 : 并确保使用关键字 lambda,其余的几乎相同。

// JavaScript Arrow Function
const square = number => number * number;

// Python Lambda Expression
square = lambda number: number * number

arrow 函数和 lambda 表达式之间的一个关键区别是,arrow 函数能够通过多个语句扩展成完整的函数,而 lambda 表达式仅限于返回的单个表达式。因此,在使用 map()、filter()或 reduce()时,如果需要对每个项执行多个操作,请先定义函数,然后再包含它。

def inefficientSquare(number):
   result = number * number
   return result

map(inefficientSquare, my_list)

替换 for 循环

好了,下面来点好东西。下面是三个常见的 for 循环示例,它们将被 map、filter 和 reduce 替换。我们的编程目标:计算列表中奇数平方和。

首先,使用 基本的 for 循环示例。注意:下面的代码纯粹是为了演示,即使没有 map/filter/reduce 也有改进空间。

numbers = [1,2,3,4,5,6]
odd_numbers = []
squared_odd_numbers = []
total = 0

# filter for odd numbers
for number in numbers:
  if number % 2 == 1:
     odd_numbers.append(number)

# square all odd numbers
for number in odd_numbers:
  squared_odd_numbers.append(number * number)

# calculate total
for number in squared_odd_numbers:
  total += number

# calculate average

让我们将每个步骤转换为这三个函数的其中之一:

from functools import reduce

numbers = [1,2,3,4,5,6]

odd_numbers = filter(lambda n: n % 2 == 1, numbers)

squared_odd_numbers = map(lambda n: n * n, odd_numbers)

total = reduce(lambda acc, n: acc + n, squared_odd_numbers)

有几个重要的语法要点要强调。

  • map()和 filter()本机可用。但是,reduce()必须从 Python 3 以上版本中的函数库导入

  • lambda 表达式是所有三个函数中的第一个参数,iterable 是第二个参数

  • reduce()的 lambda 表达式需要两个参数:累加器(传递给每个元素的值)和单个元素本身

记住,for 循环在代码中确实是很重要的,但是扩展工具包从来都不是坏事。

via:https://medium.com/better-programming/how-to-replace-your-python-for-loops-with-map-filter-and-reduce-c1b5fa96f43a

雷锋网雷锋网(公众号:雷锋网)雷锋网

雷锋网版权文章,未经授权禁止转载。详情见转载须知


Python编程技巧:如何用Map, Filter, Reduce代替For循环?-中国科技新闻网

https://www.leiphone.com/news/201912/r0Lrx6k2YjjqKfQF.html