python - Python 中的函数式中缀实现

Question

我有一个这样的当前实现:

class Infix(object):
  def __init__(self, func):
    self.func = func
  def __or__(self, other):
    return self.func(other)
  def __ror__(self, other):
    return Infix(partial(self.func, other))
  def __call__(self, v1, v2):
    return self.func(v1, v2)

@Infix
def Map(data, func):
  return list(map(func,data))

这很好，它按预期工作，但我还想扩展此实现以允许仅左侧解决方案。如果有人可以展示解决方案和解释，那就太好了。

这是我想做的一个例子...

 valLabels['annotations'] \
    |Map| (lambda x: x['category_id']) \
    |Unique|

Unique 定义如下...

@Infix
def Unique(data):
    return set(data)

谢谢!

Answer 1

如果你不介意删除最后的|，你可以这样做

class InfixR(object):
  def __init__(self, func):
    self.func = func
  def __ror__(self, other):
    return self.func(other)
  def __call__(self, v1):
    return self.func(v1)

@InfixR
def Unique(data):
  return set(data)

那么你的表情会是这样

valLabels['annotations'] \
    |Map| (lambda x: x['category_id']) \
    |Unique

---

您原来的 Infix 类(技术上)滥用了 bitwise or 运算符:|Map| 没什么特别的，它只是 值(value) | map | my_lambda，列表、对象和 lambda 的“按位或”，删除了几个空格，并插入了一些换行符(使用 \ 来防止解释器尝试分别处理每一行)。

在自定义类中，您可以使用__double_underscore__ 方法实现许多常用运算符，在按位或的情况下，它们是__or__ 和__ror__。

当 python 解释器遇到 | 运算符时，它首先查看右边的对象，看它是否有 __or__ 方法。然后调用 left.__or__(right)。如果未定义或返回 NotImplemented，它会查看右侧的对象，对于 __ror__(反向或)，并调用 right.__ror__(left) 。

另一部分是装饰器符号。

当你说

@Infix
def Unique(data):
    return set(data)

解释器将其扩展为

def Unique(data):
    return set(data)

Unique = Infix(Unique)

所以你得到一个 Infix 实例，它有 __or__ 和 __ror__ 方法，以及一个 __call__ 方法。您可能会猜到，当您调用 my_oby() 时，将调用 my_obj.__call__()。