python - 动态地从列表-字典混合列表中获取值

Question

我目前正在尝试获取从 MongoDB 查询接收到的 JSON 文档的某些字段的值。查询返回一个字典，通过在查询中使用 .values()，我得到了一个值列表。但是，此列表的某些值是另一个包含字典的列表。我试图找出最好的方法来动态获取列表中已有值的列表以及子列表中找到的字典的值。

例如，我有一个名为 text_list 的列表:

>>> text_list 
[[{u'field1': u'field1_value1', u'field2': u'field2_value1'},
  {u'field2': u'field2_value2'}],
 u'value1',
 u'value2']

我只想从这个列表中获取值:field1_value1, field2_value1, field2_value2, value1, value2(然后将它们放在一个新列表中，或者只是将这些值连接成一个大字符串) .

我想对多个列表执行此过程，但有时列表有这些额外的词典，有时则没有。什么是最好的(最少的计算密集型/最快的)方法来获取我正在寻找的值

我想到了以下方法:

def concatenate_list(inList):
    outString = '' #empty string that the values will be added to
    for item in inList:
        if type(item) == list:
            for i in np.arange(len(item)):
                for subitem in item[i].values():
                    outString = outString+' '+subitem
        else:
            outString = outString+' '+item
    return outString

>>> concatenate_list(text_list)
u'field1_value1 field2_value1 field2_value2 value1 value2'

这给了我正在寻找的结果，但必须有比这更好的方法。当我必须在数千个列表上运行此函数时，我认为这不是很有效。

Answer 1

你可以写一个递归函数，它可以打印值，前提是它们既不是列表也不是字典，就像这样

>>> def rec_fetcher(obj):
...     if isinstance(obj, list):
...         for item in obj:
...             yield from rec_fetcher(item)
...     elif isinstance(obj, dict):
...         for item in obj:
...             yield from rec_fetcher(obj[item])
...     else:
...         yield obj
... 
>>> list(rec_fetcher(data))
['field1_value1', 'field2_value1', 'field2_value2', 'value1', 'value2']

注意:字典是无序的。因此字典中的值可能与传递的文字中的值不在同一位置。

---

yield from 是 Python 3.x 的东西。如果您使用的是 Python 2.7，则可以简单地迭代递归调用并生成数据，如下所示

>>> def rec_fetcher(obj):
...     if isinstance(obj, list):
...         for item in obj:
...             for value in rec_fetcher(item):
...                 yield value
...     elif isinstance(obj, dict):
...         for item in obj:
...             for value in rec_fetcher(obj[item]):
...                 yield value
...     else:
...         yield obj
... 
>>> list(rec_fetcher(data))
[u'field2_value1', u'field1_value1', u'field2_value2', u'value1', u'value2']