如何实现仅生成具有列表类型值的键的字典键生成器？

Question

我正在尝试编写一个字典键生成器，仅当键映射的值是列表类型时才会生成键迭代器，使用递归和简单的isinstance 检查（我添加了一些打印语句我自己的利益，因为我似乎无法让这个东西运行）：

def list_key_generator(inputDict):
    for key in inputDict.keys():
        if (isinstance(inputDict[key], dict)): 
            print ("Entering dict: %s " % key)
            list_key_generator(inputDict[key])
        elif (isinstance(inputDict[key], list)):
            print ("Value is a list! key, value = %s, %s" % (key, inputDict[key]))
            yield key

我一定是误解了yield（不排除其他东西），因为打印出键的递归函数完全相同：

def print_list_keys(inputDict):
    for key in inputDict.keys():
        if (isinstance(inputDict[key], list)):
            print ("Value is a list! key, value = %s, %s" % (key, inputDict[key]))
        elif (isinstance(inputDict[key], dict)): 
            print_list_keys(inputDict[key])

整个示例如下所示：

#!/usr/bin/python

def print_list_keys(inputDict):
    for key in inputDict.keys():
        if (isinstance(inputDict[key], list)):
            print ("Value is a list! key, value = %s, %s" % (key, inputDict[key]))
        elif (isinstance(inputDict[key], dict)): 
            print_list_keys(inputDict[key])

def list_key_generator(inputDict):
    for key in inputDict.keys():
        if (isinstance(inputDict[key], dict)): 
            print ("Entering dict: %s " % key)
            list_key_generator(inputDict[key])
        elif (isinstance(inputDict[key], list)):
            print ("Value is a list! key, value = %s, %s" % (key, inputDict[key]))
            yield key

def main(): 

    inputDict = { "A" : { "x" : [0,1,2] , "y" : ["one", "two"] }, "B" : { "z" : ["one"] }, "C" : [9,8,7] } 

    print("Input dict : %s " % inputDict)

    # Print list keys  
    print("print_list_keys : BEGIN")
    print_list_keys(inputDict)
    print("print_list_keys : END")

    print("list_key_generator : BEGIN")
    # Print all keys in the dict, whose values are lists.
    for leafValue in list_key_generator(inputDict):
        print (leafValue)
    print("list_key_generator : END")

if (__name__ == "__main__"):
    main()

我使用的是 Python 3.4.1，输出如下：

Input dict : {'A': {'x': [0, 1, 2], 'y': ['one', 'two']}, 'C': [9, 8, 7], 'B': {'z': ['one']}} 
print_list_keys : BEGIN
Value is a list! key, value = x, [0, 1, 2]
Value is a list! key, value = y, ['one', 'two']
Value is a list! key, value = C, [9, 8, 7]
Value is a list! key, value = z, ['one']
print_list_keys : END
list_key_generator : BEGIN
Entering dict: A 
Value is a list! key, value = C, [9, 8, 7]
C
Entering dict: B 
list_key_generator : END

Answer 1

当您递归调用您的生成器时，您实际上不会迭代它们的结果。使用yield from 委托递归调用：

yield from list_key_generator(inputDict[key])

其他提示：

• 不要打电话给.keys()；您可以通过遍历字典本身来迭代键：

  for key in inputDict:
  

• 在 Python 中，使用 if 进行测试时，您不需要在表达式周围加上括号：

  if isinstance(inputDict[key], dict): 
  

您可以使用dict.items() 来获取键和值：

def list_key_generator(d):
    for key, value in d.items():
        if isinstance(value, dict): 
            print("Entering dict:", key)
            yield from list_key_generator(value)
        elif isinstance(value, list):
            print("Value is a list! key, value = {}, {}".format(key, value))
            yield key