主页 / 问题 / 1190940
Accepted
Danis
Asked:2021-10-16 04:58:56 +0000 UTC2021-10-16 04:58:56 +0000 UTC

__add__ 方法没有 if

  • 772

决定上课Vec2：

class Vec2():
    def __init__(self, x = 0, y = 0):
        self.x, self.y = x, y
    
    def __add__(self, num):
        if type(num) == Vec2:
            return Vec2(self.x + num.x, self.y + num.y)
        return Vec2(self.x + num, self.y + num)
    
    def __str__(self):
        return f"{self.x}  {self.y}"

还有一个问题，我可以实现一个__add__没有if'a的方法吗？

像这样的 C++ 代码：

vec2 operator+(vec2& v) {
    return vec2(x + v.x, y + v.y);
}
vec2 operator+(double s) {
    return vec2(x + s, y + s);
}
python

3 个回答

  1. Best Answer

    使用fastcoretypedispatch库中的装饰器：

    from fastcore.dispatch import typedispatch


class _Vec2:
    def __init__(self, x=0, y=0):
        self.x, self.y = x, y

    def __str__(self):
        return f"{self.x}  {self.y}"


class Vec2(_Vec2):
    @typedispatch
    def __add__(self, num: _Vec2):
        return Vec2(self.x + num.x, self.y + num.y)
    
    @typedispatch
    def __add__(self, num):
        return Vec2(self.x + num, self.y + num)


print(Vec2(1, 1) + 2)
print(Vec2(1, 1) + Vec2(1, 2))

    结论：

    3  3
2  3

    我不得不分配一个单独的类_Vec2，因为如果你在注解中指定你自己的类，它开始对一个未声明的符号发誓Vec2（如果你使用字符串注解"Vec2"或 add from __future__ import annotations，那么里面会抛出一个错误fastcore.dispatch）。

    fastcore 库概述： fastcore：一个被低估的 Python 库

    使用singledispatchmethod来自 Python 标准库的实现（根据我的口味，一个不太漂亮的选项，因为您需要将方法名称更改为下划线，但不需要额外的模块）：

    from functools import singledispatchmethod


class _Vec2:
    def __init__(self, x=0, y=0):
        self.x, self.y = x, y

    def __str__(self):
        return f"{self.x}  {self.y}"


class Vec2(_Vec2):
    @singledispatchmethod
    def __add__(self, num):
        return Vec2(self.x + num, self.y + num)
    
    @__add__.register
    def _(self, num: _Vec2):
        return Vec2(self.x + num.x, self.y + num.y)


print(Vec2(1, 1) + 2)
print(Vec2(1, 1) + Vec2(1, 2))
    • 4

  2. 作为一种选择：

    class Vec2():
    def __init__(self, x = 0, y = 0):
        self.x, self.y = x, y
    
    def __add__(self, num):
        return Vec2(self.x + getattr(num, 'x', num), self.y + getattr(num, 'y', num))
    
    def __str__(self):
        return f"{self.x}  {self.y}"

    但这适用于任何具有成员x或y.

    一般来说，为什么有必要？

    • 2

  3. 也是一个选项，但使用if-th：

        def __add__(self, num):
        dx = num.x if type(num) == Vec2 else num
        dy = num.y if type(num) == Vec2 else num
        return Vec2(self.x + dx, self.y + dy)
    • 0

相关问题