Python - przeciążanie operatorów

Question

Witam, od niedawna uczę się języka python ( poza C++ ), napisałem prostą testową klasę:

class Point:

    def __init__(self, name, x, y):
        self.name = name
        self.x = x
        self.y = y

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

    def __eq__(self, other):
        if self.x == other.x and self.y == other.y:
            return True
        return False

    def __add__(self, other):
        return Point("none", self.x + other.x, self.y + other.y)

    def __sub__(self, other):
        return Point("none", self.x - other.x, self.y - other.y)

    def multiplication_by_number(self, n):
        return Point(self.name, self.x * n, self.y * n)

Moje pytanie dotyczy przeciążeń metod dla różnych typów, wiem jak np. zrobić metodę która pomnoży przez siebie dwa punkty ( __mult__ ), co jednak jeśli chcę zrobić przeciążenie mnożenia punktu przez liczbę typu float a nie obiektu tej samej klasy? Tzn:

p3 = p3 * 10

bo na razie umiem to zrobić tylko tak:

p3 = p3.multiplication_by_number(10)

w c++ obie metody wyglądały by w ten sposób:

Point operator*(Point p);
Point operator*(int n);

używając C++ mogę zrobić własną metodę czy niezależną funkcję dla każdego typu obiektów... ale w pythonie typy dobierane są automatycznie, czy jest w ogóle możliwość zrobienia czegoś takiego?

Z góry dziękuje za pomoc i pozdrawiam ;)

Answer 1

Sprawdź po prostu typy danych, które przekazujesz do metody. Dodatkowo, aby zachować przemienność mnożenia, skorzystaj z metody __rmul__:

class Point:

    def __init__(self, x, y):
        self.x = x
        self.y = y

    def __mul__(self, other):
        if type(other) == type(self):
            return Point(self.x*other.x, self.y*other.y)
        if type(other) == int or type(other) == float:
            return Point(other*self.x, other*self.y)

    def __rmul__(self, other):
        return self.__mul__(other)

    def __str__(self):
        return f'P({self.x}, {self.y})'


if __name__ == "__main__":
    p1 = Point(4, 6)
    p2 = Point(5, 2)

    print(p1)
    print(p2)
    print("")
    print(p1*p2)
    print("")
    print(p1*3)
    print(3*p1)
    print("")
    print(p2*2.56)
    print(2.56*p2)

Comments

zamiast:

if type(other) == int or type(other) == float:

lepiej użyć:

if isinstance(other, float):

isinstance sprawdza polimorficznie subklasy, porównywanie type nie

---

Dzięki