转置字典(从字典列表中提取一个键的所有值)

时间:2012-09-06 13:12:27

标签: python optimization dictionary

我有一个这样的词典列表:

 data = [{'x': 1, 'y': 10},
         {'x': 3, 'y': 15},
         {'x': 2, 'y': 1},
          ... ]

我有一个功能(例如matplotlib.axis.plot),需要xy值的列表。所以我必须“转置”字典“。

第一个问题:你怎么称呼此操作? “转置”正确的术语吗?

我试过这个,但我正在寻找一种有效的方式(也许有一些特殊的numpy功能):

x = range(100)
y = reversed(range(100))
d = [dict((('x',xx), ('y', yy))) for (xx, yy) in zip(x,y)]
# d is [{'y': 99, 'x': 0}, {'y': 98, 'x': 1}, ... ]

timeit.Timer("[dd['x'] for dd in d]", "from __main__ import d").timeit()
# 6.803985118865967

from operator import itemgetter
timeit.Timer("map(itemgetter('x'), d)", "from __main__ import d, itemgetter").timeit()
# 7.322326898574829

timeit.Timer("map(f, d)", "from __main__ import d, itemgetter; f=itemgetter('x')").timeit()
# 7.098556041717529

# quite dangerous
timeit.Timer("[dd.values()[1] for dd in d]", "from __main__ import d").timeit()
# 19.358459949493408

有更好的解决方案吗?我的疑问是:在这些情况下,每次重新计算字符串'x'的哈希值?

3 个答案:

答案 0 :(得分:1)

从此answer

中窃取表单
import timeit
from operator import itemgetter
from itertools import imap

x = range(100)
y = reversed(range(100))
d = [dict((('x',xx), ('y', yy))) for (xx, yy) in zip(x,y)]
# d is [{'y': 99, 'x': 0}, {'y': 98, 'x': 1}, ... ]
D={x:y for x,y in zip(range(10),reversed(range(10)))}


def test_list_comp(d):
    return [dd['x'] for dd in d]

def test_list_comp_v2(d):
    return [(x["x"], x["y"]) for x in d]

def testD_keys_values(d):
    return d.keys()

def test_map(d):
    return map(itemgetter('x'), d)

def test_positional(d):
    return [dd.values()[1] for dd in d]

def test_lambda(d):
    return list(imap(lambda x: x['x'], d))

def test_imap_iter(d):
    return list(imap(itemgetter('x'), d))

for test in sorted(globals()):
    if test.startswith("test_"):
        print "%30s : %s" % (test, timeit.Timer("f(d)", 
              "from __main__ import %s as f, d" % test).timeit())
for test in sorted(globals()):
    if test.startswith("testD_"):
        print "%30s : %s" % (test, timeit.Timer("f(D)", 
              "from __main__ import %s as f, D" % test).timeit())

给出以下结果:

    test_imap_iter : 8.98246016151
       test_lambda : 15.028239837
    test_list_comp : 5.53205787458
 test_list_comp_v2 : 12.1928668102
          test_map : 6.38402269826
   test_positional : 20.2046790578
 testD_keys_values : 0.305969839705

显然,最大的胜利是让您的数据格式更接近您已经需要的格式,但您可能无法控制它。

就名字而言,我称之为转型。

答案 1 :(得分:0)

如果您只需要迭代值,可以考虑使用此方法:

imap(lambda x: x['x'], d)

答案 2 :(得分:0)

为什么不是这样的?

[(x["x"], x["y"]) for x in d]

将返回包含x和y位置的元组列表。我不确定它的速度,但它会摆脱lambda开销。