为了账号安全,请及时绑定邮箱和手机立即绑定

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

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

慕慕森 2019-04-08 11:18:24
有一个数据表idfidtitle1-1python2-1ruby3-1php4-1lisp51flask61django71webpy82rails93zend106dblog这是一个栏目表,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有人回答大概如下:frompprintimportpprintl=[]entries={}forid,fid,titleint:entries[id]=entry={'id':id,'fid':fid,'title':title}iffid==-1:l.append(entry)else:parent=entries[fid]parent.setdefault('son',[]).append(entry)pprint(l)
查看完整描述

2 回答

?
ibeautiful

TA贡献1993条经验 获得超5个赞

Stackoverflow上的那个答案,要t中的父元素出现在子元素之前,如果不能保证,可以用下面的方式:
fromitertoolsimportgroupby
fromoperatorimportitemgetterasget
frompprintimportpprint
#groupbyfid
tmp=dict([(k,list(rows))fork,rowsingroupby(sorted(t,key=get(1)),get(1))])
defmap_fun(row):
item=dict(zip(('id','fid','title'),row))
ifrow[0]intmp:
item['son']=find_children(row[0])
returnitem;
deffind_children(parent):
returnmap(map_fun,tmp[parent])
pprint(find_children(-1))
关于pythonic,可以试试这个>>>importthis
PS:我不是大神啊。
                            
查看完整回答
反对 回复 2019-04-08
  • 2 回答
  • 0 关注
  • 471 浏览
慕课专栏
更多

添加回答

举报

0/150
提交
取消
意见反馈 帮助中心 APP下载
官方微信