INTRODUCTION
FLOW CONTROL
FUNCTIONS
DATATYPES
OBJECT & CLASS
Bài tập Python: Lập trình hướng đối tượng (OOP) trong Python Lập trình hướng đối tượng trong Python Class Variables trong Python Tìm hiểu về Methods trong Python Cách sử dụng phương thức __init__() trong Python Các biến Instance trong Python Tìm hiểu về Class Attributes trong Python Hàm Static Methods trong Python Phương thức __str__ trong Python Phương thức __repr__ trong Python Phương thức eq trong Python Tìm hiểu phương thức __hash__ trong Python Phương thức __bool__ trong Python Phương thức del trong Python Tìm hiểu về lớp Property trong Python Tìm hiểu về nạp chồng toán tử trong Python Trình Decorator Property trong Python Thuộc tính chỉ đọc trong Python Thuộc tính Delete trong Python Sử dụng super() trong Python Sử dụng __slots__ trong Python Cách sử dụng Protocol trong Python Sử dụng Enum aliases và @enum.unique trong Python Tùy chỉnh và mở rộng lớp Enum trong Python Cách sử dụng hàm Auto() của Python Single Responsibility Principle trong Python Nguyên tắc Đóng-Mở trong Python Nguyên tắc thay thế Liskov - LSP trong Python Interface Segregation Principle - ISP trong Python. Nguyên tắc đảo ngược sự phụ thuộc trong Python Đa kế thừa trong Python Tìm hiểu về các lớp mixin trong Python Mô tả Descriptors trong Python Phân biệt Data Descriptor và Non-data Descriptor trong Python Phương thức __new__ trong Python Tìm hiểu về Class Type trong Python Lớp Metaclass trong Python Ví dụ sử dụng metaclass trong Python Tìm hiểu về decorator dataclass trong Python Tìm hiểu về các ngoại lệ trong Python Ngoại lệ Raise trong Python Sử dụng câu lệnh raise from trong Python Ngoại lệ tùy chỉnh trong Python Module trong Python Package trong Python Class trong Python Hàm khởi tạo trong Python Kế thừa trong Python Đa kế thừa trong Python Setter và Getter trong Python Override trong Python Interface trong Python Bài tập Python: Module và Class
ADVANCED TOPICS
BỔ SUNG
PYTHON CĂN BẢN
CÁC CHỦ ĐỀ
BÀI MỚI NHẤT
MỚI CẬP NHẬT

Thông báo: Download 4 khóa học Python từ cơ bản đến nâng cao tại đây.

Tìm hiểu về nạp chồng toán tử trong Python

Trong Python, nạp chồng toán tử (operator overloading) là một kỹ thuật cao cho phép bạn mở rộng các toán tử mặc định như +, -, *, += để chúng hoạt động với các đối tượng do bạn định nghĩa. Điều này giúp mã nguồn trở nên ngắn gọn, trực quan và dễ hiểu hơn khi xử lý các đối tượng phức tạp. Trong bài viết này, bạn sẽ tìm hiểu cách nạp chồng toán tử để đối tượng của bạn có thể tương tác với các toán tử có sẵn một cách tự nhiên và linh hoạt.

test php

banquyen png
Bài viết này được đăng tại freetuts.net, không được copy dưới mọi hình thức.

Giới thiệu về nạp chồng toán tử trong Python

Giả sử bạn có một lớp Point2D biểu diễn tọa độ x và y của một điểm trong không gian 2D như sau:

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

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

Để cộng hai đối tượng Point2D, bạn có thể định nghĩa một phương thức add() như sau:

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

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

    def add(self, point):
        if not isinstance(point, Point2D):
            raise ValueError('Đối tượng được cộng phải là một instance của Point2D')

        return Point2D(self.x + point.x, self.y + point.y)

Phương thức add() này sẽ kiểm tra xem đối tượng truyền vào có phải là một instance của lớp Point2D hay không. Nếu đúng, nó sẽ trả về một đối tượng Point2D mới với các giá trị x và y là tổng của hai điểm.

Bài viết này được đăng tại [free tuts .net]

Dưới đây là ví dụ sử dụng phương thức add() để cộng hai điểm:

a = Point2D(10, 20)
b = Point2D(15, 25)
c = a.add(b)

print(c)  # Output: (25,45)

Mặc dù cách này hoạt động tốt, nhưng Python cung cấp một cách tốt hơn là sử dụng toán tử +. Bạn có thể nạp chồng toán tử + để thay thế cho phương thức add():

c = a + b

Khi bạn sử dụng toán tử + trên đối tượng Point2D, Python sẽ gọi phương thức đặc biệt __add__() trên đối tượng đó. Hai cách gọi sau đây là tương đương:

c = a + b
c = a.__add__(b)

Phương thức __add__() phải trả về một instance mới của đối tượng Point2D.

Khả năng sử dụng các toán tử có sẵn trên các kiểu dữ liệu tùy chỉnh được gọi là nạp chồng toán tử.

Dưới đây là cách nạp chồng toán tử + trong lớp Point2D:

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

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

    def __add__(self, point):
        if not isinstance(point, Point2D):
            raise ValueError('Đối tượng được cộng phải là một instance của Point2D')

        return Point2D(self.x + point.x, self.y + point.y)

if __name__ == '__main__':
    a = Point2D(10, 20)
    b = Point2D(15, 25)
    c = a + b
    print(c)  # Output: (25,45)

Các phương thức đặc biệt cho nạp chồng toán tử trong Python

Dưới đây là bảng liệt kê các toán tử và phương thức đặc biệt tương ứng:

Toán tử Phương thức đặc biệt
+ __add__(self, other)
__sub__(self, other)
* __mul__(self, other)
/ __truediv__(self, other)
// __floordiv__(self, other)
% __mod__(self, other)
** __pow__(self, other)
>> __rshift__(self, other)
<< __lshift__(self, other)
& __and__(self, other)
| __or__(self, other)
^ __xor__(self, other)

Ví dụ, để hỗ trợ phép trừ giữa hai điểm, bạn có thể định nghĩa phương thức __sub__() như sau:

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

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

    def __add__(self, point):
        if not isinstance(point, Point2D):
            raise ValueError('Đối tượng phải là instance của Point2D')

        return Point2D(self.x + point.x, self.y + point.y)

    def __sub__(self, other):
        if not isinstance(other, Point2D):
            raise ValueError('Đối tượng phải là instance của Point2D')

        return Point2D(self.x - other.x, self.y - other.y)

if __name__ == '__main__':
    a = Point2D(10, 20)
    b = Point2D(15, 25)
    c = b - a
    print(c)  # Output: (5,5)

Nạp chồng toán tử nội tại trong Python

Một số toán tử có phiên bản nội tại, ví dụ, phiên bản nội tại của ++=.

  • Đối với các kiểu dữ liệu bất biến (immutable) như tuple, string, số, các toán tử nội tại thực hiện phép tính và không gán lại kết quả cho đối tượng ban đầu.
  • Đối với các kiểu dữ liệu có thể thay đổi (mutable), toán tử nội tại cập nhật trực tiếp trên đối tượng ban đầu mà không cần gán lại.

Python cũng cung cấp danh sách các phương thức đặc biệt để nạp chồng toán tử nội tại:

Toán tử Phương thức đặc biệt
+= __iadd__(self, other)
-= __isub__(self, other)
*= __imul__(self, other)
/= __itruediv__(self, other)
//= __ifloordiv__(self, other)
%= __imod__(self, other)
**= __ipow__(self, other)
>>= __irshift__(self, other)
<<= __ilshift__(self, other)
&= __iand__(self, other)
|= __ior__(self, other)
^= __ixor__(self, other)

Ví dụ, để nạp chồng toán tử +=, giả sử bạn có một giỏ hàng (Cart) và muốn thêm sản phẩm vào giỏ hàng như sau:

cart += item

Để hỗ trợ toán tử +=, bạn cần định nghĩa phương thức đặc biệt __iadd__() trong lớp Cart.

Đầu tiên, định nghĩa lớp Item với các thuộc tính name, qty (số lượng), và price. Lớp này có một thuộc tính amount trả về thành tiền của sản phẩm:

class Item:
    def __init__(self, name, qty, price):
        self.name = name
        self.qty = qty
        self.price = price

    @property
    def amount(self):
        return self.qty * self.price

    def __str__(self):
        return f'{self.name} {self.qty} ${self.price} ${self.amount}'

Tiếp theo, định nghĩa lớp Cart và triển khai phương thức __iadd__():

class Cart:
    def __init__(self):
        self.items = []

    def __iadd__(self, item):
        if not isinstance(item, Item):
            raise ValueError('Sản phẩm phải là một instance của Item')

        self.items.append(item)
        return self

    @property
    def total(self):
        return sum([item.amount for item in self.items])

    def __str__(self):
        if not self.items:
            return 'Giỏ hàng rỗng'

        return '\n'.join([str(item) for item in self.items])

Cuối cùng, sử dụng toán tử += để thêm sản phẩm vào giỏ hàng:

if __name__ == '__main__':
    cart = Cart()

    cart += Item('Táo', 5, 2)
    cart += Item('Chuối', 20, 1)
    cart += Item('Cam', 10, 1.5)

    print(cart)
    print('-' * 30)
    print('Tổng cộng: $', cart.total)

Kết quả đầu ra:

Táo   5    $2    $10
Chuối 20   $1    $20
Cam   10   $1.5  $15.0
------------------------------
Tổng cộng: $ 45.0

Kết bài

Nạp chồng toán tử giúp tăng cường tính linh hoạt và trực quan cho các lớp tùy chỉnh trong Python, cho phép chúng hoạt động mượt mà với các toán tử có sẵn. Bằng cách sử dụng nạp chồng toán tử, bạn có thể làm cho mã nguồn dễ đọc hơn và thuận tiện hơn trong việc thao tác với các đối tượng phức tạp. Việc này không chỉ cải thiện hiệu suất lập trình mà còn giúp bạn viết mã sạch, rõ ràng và dễ bảo trì.

Cùng chuyên mục:

Sử dụng câu lệnh raise from trong Python

Sử dụng câu lệnh raise from trong Python

Ngoại lệ tùy chỉnh trong Python

Ngoại lệ tùy chỉnh trong Python

Ngoại lệ Raise trong Python

Ngoại lệ Raise trong Python

Tìm hiểu về các ngoại lệ trong Python

Tìm hiểu về các ngoại lệ trong Python

Tìm hiểu về decorator dataclass trong Python

Tìm hiểu về decorator dataclass trong Python

Ví dụ sử dụng metaclass trong Python

Ví dụ sử dụng metaclass trong Python

Lớp Metaclass trong Python

Lớp Metaclass trong Python

Tìm hiểu về Class Type trong Python

Tìm hiểu về Class Type trong Python

Phương thức __new__ trong Python

Phương thức __new__ trong Python

Phân biệt Data Descriptor và Non-data Descriptor trong Python

Phân biệt Data Descriptor và Non-data Descriptor trong Python

Mô tả Descriptors trong Python

Mô tả Descriptors trong Python

Tìm hiểu về các lớp mixin trong Python

Tìm hiểu về các lớp mixin trong Python

Đa kế thừa trong Python

Đa kế thừa trong Python

Nguyên tắc đảo ngược sự phụ thuộc trong Python

Nguyên tắc đảo ngược sự phụ thuộc trong Python

Interface Segregation Principle - ISP trong Python.

Interface Segregation Principle - ISP trong Python.

Nguyên tắc thay thế Liskov - LSP trong Python

Nguyên tắc thay thế Liskov - LSP trong Python

Nguyên tắc Đóng-Mở trong Python

Nguyên tắc Đóng-Mở trong Python

Single Responsibility Principle trong Python

Single Responsibility Principle trong Python

Cách sử dụng hàm Auto() của Python

Cách sử dụng hàm Auto() của Python

Tùy chỉnh và mở rộng lớp Enum trong Python

Tùy chỉnh và mở rộng lớp Enum trong Python

Top