繁体   English   中英

获取嵌套字典中所有二阶键的列表

[英]Get list of all second-order keys in nested dictionary

从我的字典词典中,我希望收到所有二阶键的列表。

myDict = {
    u'A': {'1998': u'ATLANTA'},
    u'B': {'1999': u'MANNHEIM'},
    u'C': {'2000': u'BERLIN'},
    u'D': {'1998': u'CHICAGO', '1999': u'PRINCETON'},
    u'E': {'2000': u'LOUISIANA'},
    u'F': {'1998': u'NEW YORK', '1999': u'NEW YORK'}
}

我做

years = []
for author in author_aff_dict:
    years.extend(author_aff_dict[author].keys())
print uniqfy(years)

其中uniqfy()来自http://www.peterbe.com/plog/uniqifiers-benchmark

def uniqfy(seq, idfun=None):
   if idfun is None:
       def idfun(x): return x
   seen = {}
   result = []
   for item in seq:
       marker = idfun(item)
       if marker in seen: continue
       seen[marker] = 1
       result.append(item)
   return result

一切都按预期工作(即years['1998', '2000', '1999'] ),但我确信必须有更好/更短的方法来获取嵌套字典的所有键列表。

您可以使用集合理解:

>>>s= {j for i in myDict.values() for j in i}
set(['1999', '1998', '2000'])

然后,如果您只想要一个列表对象,可以使用list()将该set转换为列表。

>>> list(s)
['1999', '1998', '2000']
>>> myList = []
>>> for i in myDict.values():
...     for j in i.keys():
...             if j not in myList: myList.append(j)
...             else: continue
... 
>>> myList
['1998', '2000', '1999']

编辑:我更喜欢@Kasra的回答:)

暂无
暂无

声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.

 
粤ICP备18138465号  © 2020-2024 STACKOOM.COM