Python: 如何根据键的值对字典进行切片？

Question

假设我有一个这样构建的字典：

d={0:1, 1:2, 2:3, 10:4, 11:5, 12:6, 100:7, 101:8, 102:9, 200:10, 201:11, 202:12}

我想通过切片 d 来创建一个子词典 d1，这样 d1 包含以下键：0, 1, 2, 100, 101, 102。最终输出应该是：

d1={0:1, 1:2, 2:3, 100:7, 101:8, 102:9}

考虑到我的真实字典包含超过 2,000,000 个项目，是否有一种有效的 Pythonic 方式来执行此操作？

我认为这个问题适用于所有的情况，其中键是整数，当切片需要遵循某些不等式规则，当最终结果需要是一堆切片时放在同一个字典里。

Answer 1

您可以使用字典理解：

d = {0:1, 1:2, 2:3, 10:4, 11:5, 12:6, 100:7, 101:8, 102:9, 200:10, 201:11, 202:12}
keys = (0, 1, 2, 100, 101, 102)
d1 = {k: d[k] for k in keys}

在 python 2.7 中，您还可以计算密钥（在 python 3.x 中将 it.ifilter(...) 替换为 filter(...)）：

import itertools as it

d = {0:1, 1:2, 2:3, 10:4, 11:5, 12:6, 100:7, 101:8, 102:9, 200:10, 201:11, 202:12}
d1 = {k: d[k] for k in it.ifilter(lambda x: 1 < x <= 11, d.keys())}

Answer 2

创建子词典的一种简洁方法是使用 operator.itemgetter。此函数采用多个参数，并且 return 是一个新函数，用于 return 包含给定可迭代对象的相应元素的元组。

from operator import itemgetter as ig

k = [0, 1, 2, 100, 101, 102]
# ig(0,1,2,100,101,102) == lambda d : (d[0], d[1], d[2], d[100], d[101], d[102])
d1 = dict(zip(k, ig(*k)(d)))

Python: 如何根据键的值对字典进行切片？

Python: how to slice a dictionary based on the values of its keys?

python

dictionary

key

slice

hashable