Ghi đè "+ =" bằng Python? (__iadd __() method)

Question

Có thể ghi đè + = bằng Python không?

Answer 1

Có, ghi đè phương thức __iadd__. Ví dụ:

def __iadd__(self, other): 
    self.number += other.number 
    return self  

Answer 2

http://docs.python.org/reference/datamodel.html#emulating-numeric-types

Ví dụ, để thực hiện báo cáo kết quả x + = y, trong đó x là một thể hiện của một lớp rằng có một phương pháp __iadd __(), x .__ iadd__ (y) được gọi.

Answer 3

Ngoài quá tải __iadd__ (nhớ tự trả lại!), Bạn cũng có thể dự phòng trên __add__, vì x + = y sẽ hoạt động như x = x + y. (Đây là một trong những cạm bẫy của + = toán tử.)

>>> class A(object): 
... def __init__(self, x): 
...  self.x = x 
... def __add__(self, other): 
...  return A(self.x + other.x) 
>>> a = A(42) 
>>> b = A(3) 
>>> print a.x, b.x 
42 3 
>>> old_id = id(a) 
>>> a += b 
>>> print a.x 
45 
>>> print old_id == id(a) 
False 

Nó thậm chí trips up experts:

class Resource(object): 
    class_counter = 0 
    def __init__(self): 
    self.id = self.class_counter 
    self.class_counter += 1 

x = Resource() 
y = Resource() 

gì giá trị gì bạn mong đợi x.id, y.id, và Resource.class_counter để có?

Answer 4

Ngoài những gì được đưa ra chính xác trong câu trả lời ở trên, cần làm rõ một cách rõ ràng rằng khi __iadd__ bị ghi đè, hoạt động x += y KHÔNG kết thúc với phương thức __iadd__.

Thay vào đó, kết thúc bằng x = x.__iadd__(y). Nói cách khác, Python gán giá trị trả về của việc triển khai __iadd__ cho đối tượng bạn đang "thêm vào", SAU KHI triển khai hoàn tất.

Điều này có nghĩa là có thể biến đổi bên trái của thao tác x += y để bước ẩn cuối cùng không thành công. Hãy xem xét những gì có thể xảy ra khi bạn đang làm tăng thêm cái gì đó là trong một danh sách:

>>> x[1] += y # x has two items

Bây giờ, nếu thực hiện __iadd__ của bạn (một phương pháp của một đối tượng ở x[1]) sai lầm hoặc về mục đích loại bỏ các mục đầu tiên (x[0]) từ đầu danh sách, Python sẽ chạy phương thức __iadd__ của bạn) & cố gắng gán giá trị trả về của nó cho x[1]. Mà sẽ không còn tồn tại (nó sẽ là x[0]), dẫn đến một ÌndexError.

Hoặc, nếu __iadd__ chèn của bạn một cái gì đó để đầu x của ví dụ trên, đối tượng của bạn sẽ có mặt tại x[2], không x[1], và bất cứ điều gì là trước đó tại x[0] bây giờ sẽ có mặt tại x[1] và được gán giá trị trả về của __iadd__ yêu cầu.

Trừ khi người ta hiểu điều gì đang xảy ra, kết quả là lỗi có thể là một cơn ác mộng cần khắc phục.