切片字典

Question

我有一本字典，想将其一部分传递给函数，该部分由键列表（或元组）给出。像这样：

# the dictionary
d = {1:2, 3:4, 5:6, 7:8}

# the subset of keys I'm interested in
l = (1,5)

现在，理想情况下，我希望能够做到这一点：

>>> d[l]
{1:2, 5:6}

…但这不起作用，因为它将查找名为的键(1,5)。而且d[1,5]甚至不是有效的Python（尽管看起来很方便）。

我知道我可以这样做：

>>> dict([(key, value) for key,value in d.iteritems() if key in l])
{1: 2, 5: 6}

或这个：

>>> dict([(key, d[key]) for key in l])

这更紧凑…但是我觉得必须有一种“更好”的方式来做到这一点。我想念一个更优雅的解决方案吗？

（我正在使用Python 2.7）

Answer 1

您应该遍历元组并检查键是否在dict中，而不是相反，如果不检查键是否存在并且它不在dict中，您将得到键错误：

print({k:d[k] for k in l if k in d})

一些时间：

 {k:d[k] for k in set(d).intersection(l)}

In [22]: %%timeit                        
l = xrange(100000)
{k:d[k] for k in l}
   ....: 
100 loops, best of 3: 11.5 ms per loop

In [23]: %%timeit                        
l = xrange(100000)
{k:d[k] for k in set(d).intersection(l)}
   ....: 
10 loops, best of 3: 20.4 ms per loop

In [24]: %%timeit                        
l = xrange(100000)
l = set(l)                              
{key: d[key] for key in d.viewkeys() & l}
   ....: 
10 loops, best of 3: 24.7 ms per

In [25]: %%timeit

l = xrange(100000)
{k:d[k] for k in l if k in d}
   ....: 
100 loops, best of 3: 17.9 ms per loop

我看不到如何{k:d[k] for k in l}不可读或不美观，如果所有元素都在d中，那么它会非常有效。