一摞Python风格的纸牌
import collections
from random import choice
Card = collections.namedtuple('Card', ['rank','suit'])
def main():
class FrenchDeck:
ranks = [str(n) for n in range(2,11)] + list('JQKA')
suits = 'spades diamonds clubs hearts'.split()
def __init__(self):
self._cards = [Card(rank,suit) for suit in self.suits for rank in self.ranks]
def __len__(self):
return len(self._cards)
def __getitem__(self, idx):
return self._cards[idx]
deck = FrenchDeck()
print(len(deck))
print(deck[0])
print(deck[-1])
print(choice(deck))
print(deck[:3])
print(deck[12::13])
for card in deck:
print(card)
for card in reversed(deck):
print(card)
suit_values = dict(spades=3, hearts=2, diamonds=1, clubs=0)
def spades_high(card):
rank_value = FrenchDeck.ranks.index(card.rank)
return rank_value*len(suit_values)+suit_values[card.suit]
for car in sorted(deck, key=spades_high):
print(car)
if __name__ == '__main__':
main()
如何使用特殊方法
- 特殊方法的存在是为了被Python解释器调用的,你自己并不需要调用他们。尽量少使用
my_object.__len__() 这种写法,而推荐使用len(my_object) 。 - 如果
my_object 是一个自定义类的对象,那么Python会自己去调用你实现的__len__ 方法。 - 如果是Python内置的类型,那么 CPython 会抄个近路,
__len__ 实际上会直接返回PyVarObject 里的ob_size 属性。
模拟数值类型
from math import hypot
def main():
class Vector:
def __init__(self, x=0, y=0):
self.x = x
self.y = y
def __repr__(self):
return 'Vector(%r, %r)'%(self.x, self.y)
def __abs__(self):
return hypot(self.x, self.y)
def __bool__(self):
return bool(abs(self))
def __add__(self, other):
x = self.x + other.x
y = self.y + other.y
return Vector(x,y)
def __mul__(self, scalar):
return Vector(self.x*scalar, self.y*scalar)
v = Vector(3,4)
print(abs(v))
v = v*3
print(v)
v1 = Vector(3,4)
v2 = Vector(1,1)
print(v1+v2)
if __name__ == '__main__':
main()
注意
__repr__ 和 __str__ 的区别在于,后者是在 str() 函数被使用, 或是在用 print 函数打印一个对象的时候才被调用的,并且它返回的 字符串对终端用户更友好。 如果你只想实现这两个特殊方法中的一个,__repr__ 是更好的选择, 因为如果一个对象没有 __str__ 函数,而 Python 又需要调用它的时 候,解释器会用 __repr__ 作为替代。
|