python 中的内建函数__len__和__getitem__
最近在看流畅的python, 把作者的第一个例子在此举例:
import collections
from random import choice
# 定义一个没有函数的类, 类名为Card, 数据成员为 rank 和suit
Card = collections.namedtuple('Card', ['rank','suit'])```
# 定义扑克牌, 并初始化, 自己实现了 __len__ 和 __getitem__
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, position):
return self._cards[position]
可以看到上面的类中实现了__len__ 和__getitem__. 那么如果我们创建FrenchDeck的对象, 那么我们对该对象使用len函数和列表访问操作, 让我们来试验一下。
# 创建对象
deck = FrenchDeck()
# 获取对象中纸牌的长度
len(deck)
结果: 52
# 打印第一个纸牌
print(deck[0])
结果: Card(rank='2', suit='spades')
# 随机选择吗, 每次运算结果不一样
print(choice(deck))
结果:Card(rank='10', suit='hearts')
支持对对对象内的纸牌排序。来试验一下。
# 对扑克牌上的不同的形状和颜色赋予不同的权重
suit_values = dict(spades=3,hearts=2, diamonds=1, clubs=0)
# 输入一张扑克牌,对扑克牌的大小和花色映射到不同的整数值, 范围应该是1~52
def spades_high(card):
rank_values = FrenchDeck.ranks.index(card.rank)
return rank_values*len(suit_values) + suit_values[card.suit]
下面按照升序排列
for card in sorted(deck, key=spades_high):
print(card)
来看一下结果
Card(rank='2', suit='clubs')
Card(rank='2', suit='diamonds')
Card(rank='2', suit='hearts')
Card(rank='2', suit='spades')
。。。。。。。
。。。。。。。
Card(rank='K', suit='clubs')
Card(rank='K', suit='diamonds')
Card(rank='K', suit='hearts')
Card(rank='K', suit='spades')
Card(rank='A', suit='clubs')
Card(rank='A', suit='diamonds')
Card(rank='A', suit='hearts')
Card(rank='A', suit='spades')
是不是很神奇。建议看看原书,加深理解。