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.).