Python 相当于 java-streams 管道

Sha*_*arg 4 python list-comprehension pipe java-stream

我主要用Java编程,我发现对于数据分析Python更方便。我正在寻找一种与java流等效的管道操作方法。例如,我想做类似的事情(我混合了java和python语法)。

(key, value) = Files.lines(Paths.get(path))
   .map(line -> new Angle(line))
   .filter(angle -> foo(angle))
   .map(angle -> (angle, cosine(angle)))
   .max(Comparator.comparing(Pair::getValue)
Run Code Online (Sandbox Code Playgroud)

在这里,我从文件中获取线列表,将每条线转换为角度对象,通过某个参数过滤角度,然后创建对列表,最后找到最大对。另外可能还有多个附加操作,但要点是这是一个将一个操作的输出传递到下一个操作的管道。

我了解 python 列表理解,但是它们似乎仅限于单个“映射”和单个“过滤器”。如果我需要使用理解来传输多个映射,则表达式很快就会变得复杂(我需要将一个理解放在另一个理解中)

python 中是否有一种语法结构允许在一个命令中添加多个操作?

jiz*_*AMA 5

自己实现并不难,例如:

class BasePipe:
    def __init__(self, data):
        self.data = data
    
    def filter(self, f):
        self.data = [d for d in self.data if f(d)]
        return self
    
    def map(self, f):
        self.data = [*map(f, self.data)]
        return self
    
    def __iter__(self):
        yield from self.data
    
    def __str__(self):
        return str(self.data)
    
    def max(self):
        return max(self.data)

    def min(self):
        return min(self.data)

value = (
    BasePipe([1, 2, 3, 4]).
    map(lambda x: x * 2).
    filter(lambda x: x > 4).
    max()
)
Run Code Online (Sandbox Code Playgroud)

并给出:

8
Run Code Online (Sandbox Code Playgroud)