摘要:我还记得第一次遇到这种情况的情形。我只是想查看对象的内容,Python 却给我一个神秘的内存地址。我完全不知道发生了什么。结果发现,Python 并没有想象中的那么难——只是我没有学会让对象如何正确地表示自己。
了解 Python 的 dunder 方法:
__str____add__、__eq__等,可以深入了解如何让你的代码更简洁、更直观、更强大。你是否曾在 Python 中打印过一个对象并得到类似这样的结果?
是的……非常有帮助,对吧?
我还记得第一次遇到这种情况的情形。我只是想查看对象的内容,Python 却给我一个神秘的内存地址。我完全不知道发生了什么。结果发现,Python 并没有想象中的那么难——只是我没有学会让对象如何正确地表示自己。
其实这就是dunder 方法发挥作用的地方。
这些双下划线函数(__str__、__repr__、__add__等等)可以让你自定义对象的行为。你能够利用他们接入 Python 内置行为并根据自己的需求进行调整的方式。如果你到现在还没有使用他们,那么相比你写Python代码还是比较困难的。下面就跟云朵君一起学习下dunder 方法吧~
Dunder(“双下划线”的缩写)方法是内置的 Python 函数,可让你控制对象的行为方式。
想要让你的对象像普通字符串一样打印吗?使用 __str__ 需要比较类的两个实例吗?使用 __eq__ 想要你的对象与能够使用 +么?使用__add__这些方法与 Python 内置类型的底层方法相同。你只需要在自己的类中重写它们即可。
这是一个简单的类:
class Book:def __init__(self, title, author):
self.title = title
self.author = author
现在,尝试打印一个实例:
b = Book("1984", "George Orwell")print(b)
你得到了什么?
这看起来是不太有用。
重写__str__和__repr__我喜欢将这两种方法视为一种用于与人类交互,一种用于调试:
__str__→ 我想要打印什么?__repr__→ 这在调试控制台中应该看起来是什么样子?以下是我通常实现它们的方式:
class Book:def __init__(self, title, author):
self.title = title
self.author = author
def __str__(self):
return f'"{self.title}" by {self.author}'
def __repr__(self):
return f'Book(title="{self.title}", author="{self.author}")'
现在,检查一下:
"1984" by George Orwell好多了。
我第一次学习这个的时候就被惊到了。居然可以让自定义对象像数字一样处理,用于加减乘除!
假设我正在创建一个Wallet类。我希望它能实现以下功能:
wallet1 = Wallet( 50 )wallet2 = Wallet( 30 )
total = wallet1 + wallet2 # 应该添加余额
时,我的类要做什么,要不然它根本不知道。重写__add__class Wallet:
def __init__(self, balance):
self.balance = balance
def __add__(self, other):
if isinstance(other, Wallet):
return Wallet(self.balance + other.balance)
return NotImplemented
现在:
w1 = Wallet(50)w2 = Wallet(30)
w3 = w1 + w2
print(w3.balance) # 80
它确实有效。
而且它并不仅限于加法 — — 你可以对减法 (__sub__)、乘法 (__mul__) 甚至比较 (__gt__这是我早期遇到的另一个问题。
假设我有两个具有相同余额的钱包实例:
w1 = Wallet(50)w2 = Wallet(50)
print(w1 == w2)
你会认为 Python 会说True,对吗?
不。它返回False。
这是因为,默认情况下,Python 不会比较对象内容——它会检查它们是否是内存中完全相同的对象。
重写__eq__class Wallet:def __init__(self, balance):
self.balance = balance
def __eq__(self, other):
return self.balance == other.balance if isinstance(other, Wallet) else False
现在:
w1 = Wallet(50)w2 = Wallet(50)
print(w1 == w2) # True
好多了。
我记得在尝试构建交易历史跟踪器时,我曾为此苦苦挣扎。我想像这样循环遍历我的交易:
class TransactionHistory:def __init__(self, transactions):
self.transactions = transactions
self.index = 0
history = TransactionHistory(["Deposit $100", "Withdraw $50", "Deposit $200"])
for transaction in history:
print(transaction)
但事实并非如此——Python 没有它:
def __init__(self, transactions):
self.transactions = transactions
self.index = 0
def __iter__(self):
return self
def __next__(self):
if self.index >= len(self.transactions):
raise StopIteration
transaction = self.transactions[self.index]
self.index += 1
return transaction
history = TransactionHistory(["Deposit $100", "Withdraw $50", "Deposit $200"])
for transaction in history:
print(transaction)
现在它的行为就像一个列表。
Deposit $100Withdraw $50
Deposit $200
Dunder 方法不仅仅是为了好玩——它们让你的类感觉像是内置的 Python 类型。
如果没有 __str__,你的对象打印出来就像垃圾一样。 如果没有 __eq__,Python 就不知道两个对象何时“相等”。 如果没有 __iter__,循环遍历对象就无法进行。它们功能强大、易于使用,一旦你开始使用它们,你就会想知道如果没有它们你该如何编码。
来源:一个数据人的自留地