什么是管道 |和 >> 在 python 中？ [复制]

Question

最近在学习apache Beam，发现一些python代码是这样的：

lines = p | 'read' >> ReadFromText(known_args.input)

  # Count the occurrences of each word.
  def count_ones(word_ones):
    (word, ones) = word_ones
    return (word, sum(ones))

  counts = (lines
            | 'split' >> (beam.ParDo(WordExtractingDoFn())
                          .with_output_types(unicode))
            | 'pair_with_one' >> beam.Map(lambda x: (x, 1))
            | 'group' >> beam.GroupByKey()
            | 'count' >> beam.Map(count_ones))

发件人：https://github.com/apache/beam/blob/master/sdks/python/apache_beam/examples/wordcount.py#L92

|和>>在python中的语法和用法是什么？

Answer 1

默认情况下，| 代表逻辑或按位 OR 运算符，>> 代表右移，但幸运的是，您可以在 Python 中重载运算符。因此，为了对 | 和 >>, 进行自定义定义，您只需在类 __or__ 和 __rshift__ 中重载以下两个 dunder(magic) 方法：

class A():
    def __or__(self):
        pass
    def __rshift__(self):
        pass

我建议您阅读有关Python Data Model 的更多信息。

现在查看 Beam Python SDK，__or__ 在 PTransform 类中被重载：

  def __or__(self, right):
    """Used to compose PTransforms, e.g., ptransform1 | ptransform2."""
    if isinstance(right, PTransform):
      return _ChainedPTransform(self, right)
    return NotImplemented