一行 Python 代码搞定一棵树

鉴客 发布于 2012/04/24 08:46
阅读 16K+
收藏 52

使用 Python 内建的 defaultdict 方法可以轻松定义一个树的数据结构。

def tree(): return defaultdict(tree) 

这就是全部,就一行代码。

简单的说树也可以是一个字典数据结构

如果你继续下面的代码,需要先引入

from collections import defaultdict

实例

JSON-esque

现在我们创建一个 JSON-esque 嵌套字典无需显式的创建子字典:

users = tree() users['harold']['username'] = 'hrldcpr' users['handler']['username'] = 'matthandlersux' 

然后可通过 print(json.dumps(users)) 来打印 JSON 数据,结果如下:

{"harold": {"username": "hrldcpr"}, "handler": {"username": "matthandlersux"}}

无需赋值

我们不需要通过赋值就可以创建结构:

taxonomy = tree() taxonomy['Animalia']['Chordata']['Mammalia']['Carnivora']['Felidae']['Felis']['cat'] taxonomy['Animalia']['Chordata']['Mammalia']['Carnivora']['Felidae']['Panthera']['lion'] taxonomy['Animalia']['Chordata']['Mammalia']['Carnivora']['Canidae']['Canis']['dog'] taxonomy['Animalia']['Chordata']['Mammalia']['Carnivora']['Canidae']['Canis']['coyote'] taxonomy['Plantae']['Solanales']['Solanaceae']['Solanum']['tomato'] taxonomy['Plantae']['Solanales']['Solanaceae']['Solanum']['potato'] taxonomy['Plantae']['Solanales']['Convolvulaceae']['Ipomoea']['sweet potato'] 

要打印有好的信息,需要转成标准的字典对象:
def dicts(t): return {k: dicts(t[k]) for k in t}

现在可通过 pprint(dicts(taxonomy)) 进行打印了:

{'Animalia': {'Chordata': {'Mammalia': {'Carnivora': {'Canidae': {'Canis': {'coyote': {},                                                                             'dog': {}}},                                                       'Felidae': {'Felis': {'cat': {}},                                                                   'Panthera': {'lion': {}}}}}}},  'Plantae': {'Solanales': {'Convolvulaceae': {'Ipomoea': {'sweet potato': {}}},                            'Solanaceae': {'Solanum': {'potato': {},                                                       'tomato': {}}}}}} 

子结构也被当作是字典对象了,而叶子节点是一个空的字典对象。

迭代

可以使用有趣的方法对树进行迭代。

例如我们解析一个动物的列表并添加到之前定义的 taxonomy 中,我们可以使用如下代码:

add(taxonomy,     'Animalia,Chordata,Mammalia,Cetacea,Balaenopteridae,Balaenoptera,blue whale'.split(',')) 

简化实现:
def add(t, keys):   for key in keys:     t = t[key]

我们仍然无需赋值:

{'Animalia': {'Chordata': {'Mammalia': {'Carnivora': {'Canidae': {'Canis': {'coyote': {},                                                                             'dog': {}}},                                                       'Felidae': {'Felis': {'cat': {}},                                                                   'Panthera': {'lion': {}}}},                                         'Cetacea': {'Balaenopteridae': {'Balaenoptera': {'blue whale': {}}}}}}},  'Plantae': {'Solanales': {'Convolvulaceae': {'Ipomoea': {'sweet potato': {}}},                            'Solanaceae': {'Solanum': {'potato': {},                                                       'tomato': {}}}}}}

结论

上面提及的这些可能用处不大,只是做了一些有意思的代码。

如果你喜欢 Python 的话,把这个当成是乐趣来理解。

英文原文OSCHINA原创翻译

加载中
0
ui(
ui(

想问个问题哦.

就是怎么订阅gist上的代码.

bygreencn
bygreencn
好像只有对你comment的gist有notification。
青萍之末
青萍之末
git? watch,fork
0
一行
一行
快两年没有写py了!看起来陌生了!其实py是很好的!
0
北落
北落
刚开始看python  但是感觉跟common lisp 和javascript有太多相同点,借鉴别人的东西太多了
Nemesis_E
Nemesis_E
好像找不到 不借鉴lisp的编程语言了 而且语言添加那些被认为是好的功能 这不是更好 不过python 很难成为函数式编程语言 对函数式编程的支持也有限
ExtremeTalk
ExtremeTalk
python比Java历史还要久
少帮主
少帮主
语言相互借鉴,python是个老家伙
0
yuzhouliu
yuzhouliu

用处还是很大的,把python的dict转化成js的dict或json,  相反把js的dict或json转化成python的dict, 都是有意义,用python做后台语言的web应用就可以用这种方式和前台交流数据了。另外我想 

from collections import defaultdict 可以写成这样 

from collections import defaultdict as tree


def tree(): return defaultdict(tree) 可以去掉了


0
sevk
sevk
ruby的hash也可以json吧
0
以手指天

python哪个版本的?还有

def tree(): 
    return defaultdict(tree)

中的defaultdict(tree)的tree是什么?

返回顶部
顶部