![](/img/trans.png)
[英]How to convert list of lists to a set in python so I can compare to other sets?
[英]How can I convert a list of maps into a map of sets in python?
在Python中,我具有以下地圖列表:
[{'CN': 'SC',
'LB': 'g1k',
'SM': 'HG1'},
{'CN': 'SC',
'LB': 'g2k',
'SM': 'HG2'},
{'CN': 'SC',
'LB': 'g3k',
'SM': 'HG3'}]
我想獲得一張這樣的地圖:
{ 'CN' : 'SC',
'LB' : ['g1k', 'g2k', 'g3k'],
'SM' : ['HG1', 'HG2', 'HG3']
}
什么是最pythonic的方式來做到這一點? 謝謝!
ps此外,我打算將新的數據結構轉換為JSON,以便在網頁中顯示。
>>> list_of_map = [{'CN': 'SC',
... 'LB': 'g1k',
... 'SM': 'HG1'},
... {'CN': 'SC',
... 'LB': 'g2k',
... 'SM': 'HG2'},
... {'CN': 'SC',
... 'LB': 'g3k',
... 'SM': 'HG3'}]
>>> from collections import defaultdict
>>> d = defaultdict(set)
>>> for map in list_of_map:
... for k,v in map.items():
... d[k].add(v)
...
>>> d
defaultdict(<type 'set'>, {'LB': set(['g3k', 'g1k', 'g2k']), 'CN': set(['SC']), 'SM': set(['HG2', 'HG3', 'HG1'])})
這不是你想要的數據結構,但它非常接近,如果你真的需要的話很容易改變。 (我沒有在這里展示它,因為我認為這是set
s而不是list
的方式)。
要回到列表:
>>> back_to_lists = {k:list(v) for k,v in d.items()}
>>> back_to_lists
{'LB': ['g3k', 'g1k', 'g2k'], 'CN': ['SC'], 'SM': ['HG2', 'HG3', 'HG1']}
甚至:
>>> back_to_lists = {k:sorted(v) for k,v in d.items()}
>>> back_to_lists
{'LB': ['g1k', 'g2k', 'g3k'], 'CN': ['SC'], 'SM': ['HG1', 'HG2', 'HG3']}
In [21]: nk=[{'CN': 'SC', 'LB': 'g1k', 'SM': 'HG1'},
...: {'CN': 'SC', 'LB': 'g2k', 'SM': 'HG2'},
...: {'CN': 'SC', 'LB': 'g3k', 'SM': 'HG3'}]
In [22]: result={}
In [23]: for x in nk:
...: for k in x:
...: result.setdefault(k,set()).add(x[k])
...:
In [24]: result
Out[24]:
{'CN': set(['SC']),
'LB': set(['g3k', 'g1k', 'g2k']),
'SM': set(['HG2', 'HG3', 'HG1'])}
要么
In [60]: nk=[{'CN': 'SC', 'LB': 'g1k', 'SM': 'HG1'},
...: {'CN': 'SC', 'LB': 'g2k', 'SM': 'HG2'},
...: {'CN': 'SC', 'LB': 'g3k', 'SM': 'HG3'}]
In [61]: {z[0][0]:set(z[1]) for z in [zip(*x) for x in zip(*[y.items() for y in [x for x in nk]])]}
Out[61]:
{'CN': set(['SC']),
'LB': set(['g3k', 'g1k', 'g2k']),
'SM': set(['HG2', 'HG3', 'HG1'])}
如果新字典的值的順序無關緊要,則可以執行以下操作:
In [1]: maps = [{'CN': 'SC',
...: 'LB': 'g1k',
...: 'SM': 'HG1'},
...: {'CN': 'SC',
...: 'LB': 'g2k',
...: 'SM': 'HG2'},
...: {'CN': 'SC',
...: 'LB': 'g3k',
...:
In [2]: from itertools import chain
In [3]: {k: set(m[k] for m in maps if k in m) for k in chain.from_iterable(maps)}
Out[3]:
{'CN': set(['SC']),
'LB': set(['g2k', 'g1k', 'g3k']),
'SM': set(['HG3', 'HG2', 'HG1'])}
我會選擇解開的,不聰明的,循環的選項..
其中d是你的dicts列表,而md是一個空的dict
for d in dicts:
for k in d:
if k in md:
md[k].add(d[k])
else:
md[k]=set([d[k]])
聲明:本站的技術帖子網頁,遵循CC BY-SA 4.0協議,如果您需要轉載,請注明本站網址或者原文地址。任何問題請咨詢:yoyou2525@163.com.