python - 如何将具有相同第一个元素的单独列表的列表列表转换为字典？

Question

我需要像这样转换列表列表:

[['S', 'NP', 'VP'], ['NP', 'Det', 'N'], ['NP', 'NP', 'PP'], ['VP', 'V', 'NP'], ['VP', 'VP', 'PP'], ['PP', 'P', 'NP'], ['Det', "'the'"], ['N', "'pirate'"], ['N', "'sailor'"], ['N', "'telescope'"], ['V', "'sees'"], ['P', "'with'"]]

字典如下所示:

{'S':['NP', 'VP'], 'NP': ['Det', 'N'], ['NP', 'PP'], 'VP': ['V', 'NP'], ['VP', 'PP'], 'PP': ['P', 'NP'], 'Det': ["'the'"], 'N': ["'pirate'"], ["'sailor'"], ["'telescope'"], 'V': ["'sees'"], 'P': ["'with'"]}

我尝试使用此方法使用 from collections import default dict:

g = defaultdict(dict)
for i, j, k in new_grammar:
    g[i][j] = k

但这行不通，因为列表列表中有列表只有两个元素。

我也试过:

grammar = {}
for rule in new_grammar:
    grammar[rule[0]] = rule[1:]

但是，这只会给每个键一个值。

有什么办法吗？

Answer 1

你走在正确的轨道上，但你的 defaultdict 是一个从字符串到列表的映射，而不是字典。试试这个:

g = defaultdict(list)
for i in new_grammar:
    g[i[0]].extend(i[1:])