python 数据结构转换_python 数据结构转换,将线性元祖转换成字典树

有一个数据表

id fid title

1 -1 python

2 -1 ruby

3 -1 php

4 -1 lisp

5 1 flask

6 1 django

7 1 webpy

8 2 rails

9 3 zend

10 6 dblog

这是一个栏目表,title 是栏目名称, id 是栏目 id, fid 是栏目的父id.构建一个多级栏目分类

通过 python 查询处理,会得到下面的一个元祖,

t = (

(1, -1, 'python'),

(2, -1, 'ruby'),

(3, -1, 'php'),

(4, -1, 'lisp'),

(5, 1, 'flask'),

(6, 1, 'django'),

(7, 1, 'webpy'),

(8, 2, 'rails'),

(9, 3, 'zend'),

(10, 6, 'dblog')

)

希望通过 python 处理,转换成下面的 list 字典树

l = [

{

'id': 1,

'fid': -1,

'title': 'python',

'son': [

{

'id': 5,

'fid': 1,

'title': 'flask',

},

{

'id': 6,

'fid': 1,

'title': 'django',

'son': [

{

'id': 10,

'fid': 6,

'title': 'dblog',

},

]

},

{

'id': 7,

'fid': 1,

'title': 'webpy',

},

]

},

{

'id': 2,

'fid': -1,

'title': 'ruby',

'son': [

{

'id': 8,

'fid': 2,

'title': 'rails',

},

]

},

{

'id': 3,

'fid': -1,

'title': 'php',

'son': [

{

'id': 9,

'fid': 3,

'title': 'zend',

},

]

},

{

'id': 4,

'fid': -1,

'title': 'lisp',

}

]

也就是类似网站的目录, 父栏目包含子栏目.

自己写了好几个,感觉效率不够好, 求大神更 pythonic 的方法

在 stackoverflow 有人回答 大概如下：

from pprint import pprint

l = []

entries = {}

for id, fid, title in t:

entries[id] = entry = {'id': id, 'fid': fid, 'title': title}

if fid == -1:

l.append(entry)

else:

parent = entries[fid]

parent.setdefault('son', []).append(entry)

pprint(l)