python：如何获取字典的子集

Question

我有一个字典，里面有很多元素。我想写一个函数，可以根据给定的索引范围返回这些元素（把字典当成数组来用）：

get_range(dict, begin, end):
    return {a new dict for all the indexes between begin and end}

这该怎么做呢？

补充说明：我不是在问怎么用键来过滤... 比如说)

{"a":"b", "c":"d", "e":"f"}

get_range(dict, 0, 1) returns {"a":"b", "c":"d"} (the first 2 elements)

我不在乎排序... 其实我是在实现服务器端的分页功能...

Answer 1

放心吧，如果你真的想要一个 OrderedDict，你也可以使用 enumerate：

#!/usr/bin/env python
def get_range(d, begin, end):
    return dict(e for i, e in enumerate(d.items()) if begin <= i <= end)

if __name__ == '__main__':
    print get_range({"a":"b", "c":"d", "e":"f"}, 0, 1)

输出结果：

{'a': 'b', 'c': 'd'}

补充说明：我让你用 0, 1 作为范围值，但其实你应该用 0, 2 来表示“前两个元素”（并且使用 begin <= i < end 作为比较函数）。

Answer 2

直接的实现方式：

def get_range(d, begin, end):
    result = {}
    for (key,value) in d.iteritems():
        if key >= begin and key <= end:
            result[key] = value
    return result

一行代码：

def get_range2(d, begin, end):
    return dict([ (k,v) for (k,v) in d.iteritems() if k >= begin and k <= end ])

Answer 3

编辑： 字典是无序的。这意味着每次你修改字典后，get_range 返回的结果可能会不一样。如果你想要每次都得到相同的结果，可以把你的 dict 换成 collections.OrderedDict。

不过，你可以使用 itertools.islice 来获取一个切片：

import itertools
def get_range(dictionary, begin, end):
  return dict(itertools.islice(dictionary.iteritems(), begin, end+1)) 

---

下面保留了之前通过键过滤的答案：

使用 @Douglas 的算法，我们可以通过生成器表达式来简化它：

def get_range(dictionary, begin, end):
  return dict((k, v) for k, v in dictionary.iteritems() if begin <= k <= end)

顺便说一下，不要把 dict 用作变量名，因为在这里 dict 是字典的构造函数。

如果你使用的是 Python 3.x，可以直接使用字典推导式。

def get_range(dictionary, begin, end):
  return {k: v for k, v in dictionary.items() if begin <= k <= end}