使用列表理解作为嵌套在字典理解中的条件过滤器

Question

我正在尝试创建一个字典的字典，它是通过过滤一个已知的字典创建的，条件来自一个列表。这将是动态的，因此列表和已知字典中的数据将不断变化。

list = {5: ('t', 'a'), 7: ('s', 'o')}

known_dict = {('s', 'a'): 105, ('s', 'e'): 2, ('s', 'h'): 5, ('t', 'a'): 21,
              ('t', 'e'): 8, ('t', 'h'): 21}

鉴于上述情况，我正在尝试创建一个字典，其中键是列表中元组的第一个字符，值是已知字典中相应的第一个字符元组。

desired_output = {s: {('s', 'a'): 105, ('s', 'e'): 2, ('s', 'h'): 5}, 
                  t: {('t', 'a'): 21, ('t', 'e'): 8, ('t', 'h')}}

我必须尝试实现的代码是：

desired_output = {
    key[0]: {
        key: value for key, value in known_dict.items() 
        if key[0] == [a for a,b in list.values()]
    } 
    for key in known_dict
}

问题是当列表理解发生时，结果是一个列表，而不是单个字符，所以我的结果是一个空字典，因为不满足条件。

如果我遗漏了什么或者我应该考虑其他方法，请告诉我。

Answer 1

以下应该有效。遍历 "list" 中的值（你应该重命名）并沿途从 "known_dict" 中收集项目，而不是相反：

desired_output = {
    v[0]: {k_: v_ for k_, v_ in known_dict.items() if k_[0]==v[0]} 
    for v in list.values()
}
# {'t': {('t', 'a'): 21, ('t', 'e'): 8, ('t', 'h'): 21}, 
#  's': {('s', 'a'): 105, ('s', 'e'): 2, ('s', 'h'): 5}}

旁注："list" 是一个糟糕的变量名。首先，它隐藏了内置类型 list。其次，您将其用于 dict.

使用列表理解作为嵌套在字典理解中的条件过滤器

Using a list comprehension used as a conditional filter nested in a dictionary comprehension

python

dictionary

list

nested-lists

dictionary-comprehension