python遍历map_Python下避免filter,map(reduce)重复遍历

Hi 第一次来写东西&＃xff0c;大家多多支持

(入题)

最近某天上班路上在微薄看到一哥们写的《在 Javascript 中实现 LINQ 》看到里面关于 C#的 Linq 在实现 filter 和 map 的时候说道(reduce 已经在 python3 从全局空间去掉了&＃xff0c;所以标题里面我加了个括号)&＃xff0c;如果同时调用类似 filter 和 map 这样的操作去遍历 List 的时候&＃xff0c;实际上只遍历了一遍&＃xff0c;像下面这样&＃xff1a;

var array &＃61; new []{ 0, 1, 2, 3, 4, 5, 6, 7, 8, 9 };

var sum &＃61; array.Where(n &＃61;> n % 2 &＃61;&＃61;&＃61; 0)

.Select(n &＃61;> n &＃43; 3)

.Aggregate((sum, n) &＃61;> sum &＃43; n, 0);

然后文章后面提到 Javascript 中直接调用 filter 和 map 的时候&＃xff0c;会重复遍历 Array &＃xff0c;比如像下面代码这样&＃xff1a;

let array &＃61; [1, 2, 3, 4]

let sum &＃61; array.filter(n &＃61;> n % 2 &＃61;&＃61;&＃61; 0)

.map(n &＃61;> n &＃43; 3)

.reduce((sum, n) &＃61;> sum &＃43; n, 0)

这样的话会先把 arrayfilter 成[2,4]&＃xff0c;然后再 map 成[5, 7]&＃xff0c;然后再 reduce 成 12 ,所以这个过程 array 会被重复遍历。

好了&＃xff0c;下面说下 Python &＃xff0c;看完文章的时候然后我就好奇 Python 里面的 filter 和 map 是不是也是这样&＃xff0c;会重复去遍历 List &＃xff0c;于是做了个实验&＃xff1a;

像平时我们喜欢的函数式的写法:

In [1]: numbers &＃61; [1,2,3,4,5,6]

In [2]: list(map(lambda x: x &＃43; 1, filter(lambda x: x % 2 &＃61;&＃61; 0, numbers)))

Out[2]: [3, 5, 7]

为了看清楚是不是重复遍历了 numbers 这个 List &＃xff0c;把 lambda 改写成个普通的 function 打印出来看看

In [3]: def is_even(number):

...: print(&＃39;filter&＃39;)

...: return True if number % 2 &＃61;&＃61; 0 else False

In [4]: def inc(number):

...: print(&＃39;map&＃39;)

...: return number &＃43; 1

In [5]: list(map(inc, filter(is_even, numbers)))

filter

filter

filter

filter

filter

filter

map

map

map

Out[6]: [3, 5, 7]

上面可以看到&＃xff0c; Python 这样直接调用 filter 和 map 也是会重复遍历 list 的。不过那哥们的文中提到后来能在 Javascript 实现 Linq &＃xff0c;主要因为 ES6 支持 yield 和 Generator Function &＃xff0c;所以我想 Python 这两个都支持肯定也是可以实现类似 Linq 这样不重复遍历的 Magic 。

然后&＃xff0c;再试了下之前很喜欢的一个函数式库 pyfunctional。这是个很值来安利一波的一个库&＃xff0c;用了这个库后&＃xff0c;摆脱了原生那种很丑的写法

# before

list(map(inc, filter(is_even, numbers)))

# afater

seq(numbers)\

.filter(is_even)\

.map(inc)\

.to_list()

嗯&＃xff0c;很 Js 的写法....

好&＃xff0c;回到正题&＃xff0c;如果像上面这样调用 functional 时&＃xff0c;发现整个过程只遍历的一次 List

In [7]: from functional import seq

In [8]: seq(numbers)\

...: .filter(is_even)\

...: .map(inc)\

...: .to_list()

filter

filter

map

filter

filter

map

filter

filter

map

Out[8]: [3, 5, 7]

果然是个好货&＃xff0c;安利一波