Python中使用flatmap

Python中使用flatmap

写多了Spark、Flink突然觉得到处是map、flatmap，在这些框架中有map、flatmap算子，在scala语言里有对应的方法，而在java 8之后，我们ye可以通过stream api使用map和flatmap，这些用法，用一次就会爱上它，代码确实优雅、好用。早之前我一直以为python里也有这样的用法，也使用过filter、map等方法，但直到今天在写一个小脚本需要用到flatmap时，才发现竟然在语言层面没有原生支持。

这里不罗逼嗦了，直接上实现flatmap的代码，参考https://dev.to/turbaszek/flat-map-in-python-3g98。

Map reduce

from functools import reduce

flat_map = lambda f, xs: reduce(lambda a, b: a + b, map(f, xs))

Map sum

flat_map = lambda f, xs: sum(map(f, xs), [])

For and extend

def flat_map(f, xs):
    ys = []
    for x in xs:
        ys.extend(f(x))
    return ys

List comprehension

flat_map = lambda f, xs: [y for ys in xs for y in f(ys)]

flat_map = lambda f, xs: (y for ys in xs for y in f(ys))