有没有办法在 Python 中为最小值和最大值返回自定义值？

Question

我有一个自定义类，

class A:
    def __init__(self, a, b):
        self.a = a
        self.b = b

该类不可迭代或可索引或类似的东西。如果可能的话，我想保持这种状态。是否有可能有类似以下的工作？

>>> x = A(1, 2)
>>> min(x)
1
>>> max(x)
2

让我想到这一点的是min 和max 在docs 中被列为“通用序列操作”。由于range 被相同的文档认为是序列类型，所以我认为必须对range 进行某种优化，也许我可以利用它。

也许有一种我不知道的神奇方法可以实现这一点？

Answer 1

是的。当min 接受一个参数时，它假定它是一个可迭代的，迭代它并取最小值。所以，

class A:
    def __init__(self, a, b):
        self.a = a
        self.b = b
    def __iter__(self):
        yield self.a
        yield self.b

应该可以。

附加说明：如果您不想使用__iter__，我不知道该怎么做。您可能想创建自己的 min 函数，如果传递给它的参数中有一个方法，则调用一些 _min_ 方法并调用旧的 min 否则。

oldmin = min
def min(*args):
    if len(args) == 1 and hasattr(args[0], '_min_'):
        return args[0]._min_()
    else:
        return oldmin(*args)

Answer 2

没有__min__ 和__max__ 特殊方法*。这是一种耻辱，因为range 已经看到了一些pretty nice optimizations in Python 3。你可以这样做：

>>> 1000000000000 in range(1000000000000)
False

但除非你想等待很长时间，否则不要尝试这个：

>>> max(range(1000000000000))

但是，正如Lærne 所建议的那样，创建自己的min/max 函数是一个不错的主意。

我会这样做。更新：根据PEP 8 的建议，删除了名称__min__ 以支持_min：

永远不要发明这样的名字；仅按文档使用它们

代码：

from functools import wraps

oldmin = min

@wraps(oldmin)
def min(*args, **kwargs)
    try:
        v = oldmin(*args, **kwargs)
    except Exception as err:
        err = err
    try:
        arg, = args
        v = arg._min()
    except (AttributeError, ValueError):
        raise err
    try:
        return v
    except NameError:
        raise ValueError('Something weird happened.')

我认为这种方式可能会更好一些，因为它可以处理其他答案没有考虑的一些极端情况。

请注意，具有_min 方法的可迭代对象仍将照常被oldmin 使用，但返回值被特殊方法覆盖。

但是，如果 _min 方法要求迭代器仍然可以使用，这将需要调整，因为迭代器首先被 oldmin 使用。

还要注意，如果__min 方法只是通过调用oldmin 来实现，那么事情仍然可以正常工作（即使迭代器已被消耗；这是因为oldmin 在这种情况下引发了ValueError）。

* 这种方法通常被称为“魔术”，但这不是首选术语。

Answer 3

由于range 被同一个文档视为序列类型，我在想range 必须进行某种优化，也许我可以利用它。

没有对范围进行优化，也没有针对min/max 的专用魔术方法。

如果您查看the implementation for min/max，您会看到在完成一些参数解析后，a call to iter(obj)（即obj.__iter__()）会获取一个迭代器：

it = PyObject_GetIter(v);
if (it == NULL) {
    return NULL;
}

然后calls to next(it)（即it.__next__）在循环中执行以获取值进行比较：

while (( item = PyIter_Next(it) )) {
    /* Find min/max  */

有没有可能做类似下面的工作？

不，如果你想使用内置的min*，你唯一的选择是实现迭代器协议。

---

*通过修补min，你当然可以让它做任何你想做的事。显然是以在 Pythonland 中运营为代价的。但是，如果您认为可以利用一些优化，我建议您创建一个 min 方法，而不是重新定义内置的 min。

此外，如果您只有整数作为实例变量并且您不介意不同的调用，您可以随时使用vars 来获取instance.__dict__，然后将它的.values() 提供给min：

>>> x = A(20, 4)
>>> min(vars(x).values())
4