时间戳

首页 > 解决方案 > 地图字典查找

python - 地图字典查找

问题描述

我有一个从字典中获取项目的功能：

def dict_lookup(dictionary, key):
    return dictionary[key]

我将通过多处理映射此函数，以从一系列非常长的 dicts 中查找值：

dictlist = [{}, {}, {}]
dict_lookup_partial = functools.partial(dict_lookup, key=some_key)
with multiprocessing.Pool() as pool:
    values = pool.map(dict_lookup_partial, dictlist)

我觉得我不应该定义一个从字典中获取值的函数。有一些内置的方法可以做到这一点吗？

标签： python

解决方案

标准库的operator模块中的itemgetter函数提供了这种行为：

>>> import multiprocessing as mp
>>> import operator
>>> dictlist = [{'a': 1, 'b':2, 'c': 10}, {'a': 3, 'b': 4, 'c': 20},
                {'a': 5, 'b': 6, 'c': 30}]
>>> agetter = operator.itemgetter('a')
>>> with mp.Pool() as pool:
...     avalues = pool.map(agetter, dictlist)
... 
>>> avalues
[1, 3, 5]

它还可以用于检索多个键的值：

>>> bcgetter = operator.itemgetter('b', 'c')
>>> with mp.Pool() as pool:
...     bcvalues = pool.map(bcgetter, dictlist)
... 
>>> bcvalues
[(2, 10), (4, 20), (6, 30)]

通常，操作员模块是寻找复制内置函数行为以用于或map的函数的第一个位置。filterreduce

推荐阅读