从字典列表中获取值

Question

sample_dict = [{'id':123, 'name':'ABC', 'loc':'XYZ'}, {'id':456, 'name':'DEF', 'loc':'ZYX'}, {'id':789, 'name':'GHI', 'loc':'YZX'}]

我想从给定的字典列表中选择键的值（ id 和 name ）。

我想要类似这样的结果：

result = ({'id': [123, 456, 789]}, {'name': ['ABC', 'DEF', 'GHI']})

除了下面列出的之外，还有其他实现吗？我想以更高的内存效率和更少的时间来实现这一点，因为我有数百万数据要解压。我们可以通过operator.itemgetter 或itertools 实现这一目标吗？哪个内存效率更高，消耗的时间更少？

使用字典和列表组合

{'id':[i.get('id') for i in sample_dict]}

{'name':[i.get('name') for i in sample_dict]}

使用地图

{'id': list(map(lambda x: x.get('id'), sample_dict))}

{'name': list(map(lambda x: x.get('name'), sample_dict))}

Answer 1

一个更简单的方法是这样的：

for i in range(len(sample_dict)):
    print(sample_dict[i]['id'])

Answer 2

new_arr = {str:list} 
for each in sample_dict:
    for keys in each.keys():
        if keys not in new_arr.keys():
            new_arr[keys] = []
        new_arr[keys].append(each[keys])

输出是{'id': [123, 456, 789], 'name': ['ABC', 'DEF', 'GHI'], 'loc': ['XYZ', 'ZYX', 'YZX']}

Answer 3

首先，您似乎在模拟数据库。为什么不实际使用数据库？或者至少是熊猫？

其次，没有办法“更有效”地迭代 1000 多个项目，因为无论如何你仍然必须迭代 1000 多个项目。是的，itertools 和 itemgetter 在每个项目的基础上可能会快得可以忽略不计，但如果您的问题是数据大小，那将无济于事。

您必须迭代整个数据集；但是，您可以这样做一次，然后就拥有一个可重用的结构。如果您发现自己针对不同的查询多次遍历整个数据集，这可以消除所有重复的工作：

from collections import defaultdict
from pprint import pprint

database = [
    {'id':123, 'name':'ABC', 'loc':'XYZ'}, 
    {'id':456, 'name':'DEF', 'loc':'ZYX'}, 
    {'id':789, 'name':'GHI', 'loc':'YZX'}]

index = defaultdict(list)
for row in database:
    for key in row:
        index[key].append(row[key])

pprint(index)

默认字典(,
{'id': [123, 456, 789],
'loc': ['XYZ', 'ZYX', 'YZX'],
'名称': ['ABC', 'DEF', 'GHI']})