为什么不能为非浮点数定义“round”？

Question

给定一个简单的类

class Vector(object):
    def __init__(self, value):
        self.value = value
    def __abs__(self):
        return math.sqrt(sum([x**2 for x in self.value]))
    def __round__(self, *n):
        return [round(x,*n) for x in self.value]

为什么abs(Vector([-3,4])) 正确产生5 而round(Vector([-3.1,4])) 抱怨TypeError: a float is required 而不是所需的[-3,4]，如何解决这个问题？

我知道round 通常应该返回一个浮点数，但是对于本例中的向量，可能的含义可能没有歧义，那么为什么不能简单地覆盖它呢？我真的必须继承numbers.Real，还是定义Vector(...).round(n)？

Answer 1

__round__ 特殊方法仅在 Python 3 中引入。Python 2 中不支持该特殊方法。

您必须使用专用方法而不是函数：

class Vector(object):
    def __init__(self, value):
        self.value = value

    def round(self, n):
        return [round(x, n) for x in self.value]

或者您必须提供自己的 round() 函数：

import __builtin__

def round(number, digits=0):
    try:
        return number.__round__(digits)
    except AttributeError:
        return __builtin__.round(number, digits)

您甚至可以将其修补到 __builtins__ 命名空间中：

import __builtin__

_bltin_round = __builtin__.round

def round(number, digits=0):
    try:
        hook = number.__round__
    except AttributeError:
        return _bltin_round(number, digits)
    else:
        # Call hook outside the exception handler so an AttributeError 
        # thrown by its implementation is not masked
        return hook(digits)

__builtin__.round = round