Sobes.tech
Junior — Middle

Hogyan lehet saját operátort definiálni Pythonban osztályokban való használatra?

sobes.tech MI

Válasz az MI-től

A Pythonban nem lehet teljesen új operátorokat létrehozni, de a meglévő operátorok viselkedését saját osztályaink számára újra definiálhatjuk speciális módszerek (varázslatos módszerek) segítségével.

Például, hogy meghatározzuk az + operátor viselkedését saját osztályunk számára, implementálnunk kell az __add__ metódust:

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

    def __add__(self, other):
        return Vector(self.x + other.x, self.y + other.y)

    def __repr__(self):
        return f"Vector({self.x}, {self.y})"

v1 = Vector(1, 2)
v2 = Vector(3, 4)
print(v1 + v2)  # Vector(4, 6)

Így nem hozol létre új operátort, hanem meghatározod, hogyan fog működni a meglévő operátor a saját osztályoddal. Ezeknek a módszereknek a teljes listáját megtalálod a Python dokumentációjában a speciális módszerek témakörében (__sub__, __mul__, __eq__, stb.).