python - 使用基于列表列表中的项目的键创建 Python 字典

Question

我正在创建一个字典，其中如果键有 'R' 则为一个列表，否则它会出现在 'E' 的列表中。到目前为止我的代码:

d1 = {'SSW 540': ['E', 'SYEN'], 'SSW 564': ['R', 'SFEN'], 'SSW 567': ['R', 'SFEN'], 'CS 501': ['E', 'SFEN']}

rlist = []
elist = []

for k, v in d1.items():
    if v[0] == 'R':
        rlist.append([v[1], k])
    elif v[0] == 'E':
        elist.append([v[1], k])

输出是:

rlist = [['SFEN', 'SSW 564'], ['SFEN', 'SSW 567']]
elist = [['SYEN', 'SSW 540'], ['SFEN', 'CS 501']]

我如何创建两个字典，将它们命名为 rdict 和 edict 其中:

rdict = {'SFEN': ['SSW 564','SSW 567']}
edict = {'SYEN': ['SSW 540'], 'SFEN': ['CS 501']}

可能有更多的值，而不仅仅是 'SYEN' 和 'SFEN'，因此它需要是列表中任何特定的第一个元素。

如果我尝试这样的事情:

mlist = {}

for list in rlist:
    if list[0] in rlist:
        mlist.append(list[1])
    else:
        mlist[list[0]].append(list[1])

我收到KeyError:'SFEN'。我可以使用defaultdict吗？

Answer 1

有一种方法可以直接从 d1 创建 m_dict。

d1 = {
    'SSW 540': ['E', 'SYEN'],
    'SSW 564': ['R', 'SFEN'],
    'SSW 567': ['R', 'SFEN'],
    'CS 501': ['E', 'SFEN'],
}

m_dict = {}
for value, (category, key) in d1.items():
    m_dict.setdefault(category, {}).setdefault(key, []).append(value)

print(m_dict)

输出:

{'E': {'SYEN': ['SSW 540'], 'SFEN': ['CS 501']}, 'R': {'SFEN': ['SSW 564', 'SSW 567']}}

说明:

• dict.setdefault(key, 默认值)
  • 如果key不存在，则按默认设置key。
  • 并且始终返回key本身的值。
• d1.items() 中的值(类别、键):
  • 您可以解压固定长度的元组、列表或其他序列。
  • 所以它相当于:

for value, tuple_ in d1.items():
    category, key = tuple_