python - 当每个键有多个值时，根据值获取键列表

Question

当每个键有多个值时，如何根据一个值获取键列表？

下面的代码可以在值为单个对象时获取列表。但是，当有多个对象作为值或值中存在其他数据时，它无法获取键。

mydict = {
    'romeo': 16,
    'juliette': 16,
    'zena': 'red', 
    'gabby': 'red', 
    'lauren': 'smart',
    'betty': 'smart',
    'kristen': 'sweet',
    'lisa': 'sweet',
    'lucy': ['smart', 'sweet'],
    'shawna': 'green field', 
    'terry' : 'the green field is on fire', 
    'dorothy': 'shelly plays on the green field',
    'jenny': ['intelligent', 'cuts the smart green field grass'], 
}

needle = 'green field'
filtered_keys = [k for k, v in mydict.items() if v == needle]
print(filtered_keys)

当needle = 16时，输出正确:['romeo', 'juliette']。

当needle = 'red'时，输出正确:['zena', 'gabby']。

当needle = 'smart'时，输出不正确:['lauren', 'betty']。
预期输出为['lauren', 'betty', 'lucy', 'jenny']。

当needle = 'sweet'时，输出不正确:['kristen', 'lisa']。
预期输出为 ['kristen', 'lisa', 'lucy']。

当needle = 'green field'时，输出不正确:['shawna']
预期输出为['shawna', 'terry', 'dorothy', 'jenny']。

Answer 1

另一个答案不处理当needle在in列表in字符串中的情况。也就是说，另一个答案仅在 needle 等于该值或直接包含在该值中时进行处理。因此，当您的问题中包含的测试用例needle = 'green field'时，另一种解决方案不起作用。

概括来说，您可以递归地检查该值是否包含 needle。

from typing import Any
from typing import Dict
from typing import Iterable

def keys_containing_value(d: Dict, needle: Any):
    def recursively_in(x: Any, n: Any):
        if isinstance(x, str) and isinstance(n, str):
            return n in x
        if type(x) is type(n):
            return x == n
        if isinstance(x, Iterable) and not isinstance(x, str):
            return any(recursively_in(item, n) for item in x)
    return [k for k, v in d.items() if recursively_in(v, needle)]

以下是该函数如何使用您的数据和测试用例工作:

mydict = {
    'romeo': 16,
    'juliette': 16,
    'zena': 'red',
    'gabby': 'red',
    'lauren': 'smart',
    'betty': 'smart',
    'kristen': 'sweet',
    'lisa': 'sweet',
    'lucy': ['smart', 'sweet'],
    'shawna': 'green field',
    'terry': 'the green field is on fire',
    'dorothy': 'shelly plays on the green field',
    'jenny': ['intelligent', 'cuts the smart green field grass'],
}
needles = [
    16,
    'red',
    'smart',
    'sweet',
    'green field',
]

for needle in needles:
    print(keys_containing_value(mydict, needle))

输出:

['romeo', 'juliette']
['zena', 'gabby']
['lauren', 'betty', 'lucy', 'jenny']
['kristen', 'lisa', 'lucy']
['shawna', 'terry', 'dorothy', 'jenny']