爲蟒蛇defaultdict

Question

什麼是定義一個自定義按鍵功能類似於向key參數list.sort，在一個collections.defaultdict使用的好辦法自定義鍵的功能？

下面是一個例子使用案例：

import collections 

class Path(object): 
    def __init__(self, start, end, *other_features): 
     self._first = start 
     self._last = end 
     self._rest = other_features 
    def startpoint(self): 
     return self._first 
    def endpoint(self): 
     return self._last 
    # Maybe it has __eq__ and __hash__, maybe not 

paths = [... a list of Path objects ...] 

by_endpoint = collections.defaultdict(list) 
for p in people: 
    by_last_name[p.endpoint()].append(p) 
# do stuff that depends on lumping paths with the same endpoint together 

我的願望是一種方法來告訴by_endpoint使用Path.endpoint爲key功能，類似於key參數list.sort，並沒有把這個鍵定義進入Path課程本身（通過__eq__和__hash__），因爲同樣支持「通過起始點結塊」也同樣明智。

Answer 1

事情是這樣的，也許：

from collections import defaultdict 

class defaultkeydict(defaultdict): 

    def __init__(self, default_factory, key=lambda x: x, *args, **kwargs): 
     defaultdict.__init__(self, default_factory, *args, **kwargs) 
     self.key_func = key 

    def __getitem__(self, key): 
     return defaultdict.__getitem__(self, self.get_key(key)) 

    def __setitem__(self, key, value): 
     defaultdict.__setitem__(self, self.get_key(key), value) 

    def get_key(self, key): 
     try: 
      return self.key_func(key) 
     except Exception: 
      return key 

注意，回落到傳入的密鑰，如果不能執行的關鍵功能的邏輯。這樣你仍然可以使用字符串或任何鍵訪問項目。

現在：

p = Path("Seattle", "Boston") 
d = defaultkeydict(list, key=lambda x: x.endpoint()) 
d[p].append(p) 
print(d)  # defaultdict(<type 'list'>, {'Boston': [<__main__.Path object at ...>]})