如何在OrderedDict的开头添加元素？

Question

我有这个：

d1 = OrderedDict([('a', '1'), ('b', '2')])

如果我这样做：

d1.update({'c':'3'})

然后我得到这个：

OrderedDict([('a', '1'), ('b', '2'), ('c', '3')])

但我想要这个：

[('c', '3'), ('a', '1'), ('b', '2')]

无需创建新字典。

Answer 1

在Python
2中没有执行此操作的内置方法。如果需要此操作，则需要编写一种以O（1）复杂度prepend()在OrderedDict内部操作的方法/函数。

对于Python 3.2和更高版本， 应
使用move_to_end方法。该方法接受一个last参数，该参数指示元素将被移动到的底部（last=True）还是顶部（last=False）OrderedDict。

最后，如果您想要一种快速，肮脏且 缓慢的 解决方案，则可以OrderedDict从头开始创建一个新的解决方案。

四种不同解决方案的详细信息：

from collections import OrderedDict

class MyOrderedDict(OrderedDict):

    def prepend(self, key, value, dict_setitem=dict.__setitem__):

        root = self._OrderedDict__root
        first = root[1]

        if key in self:
            link = self._OrderedDict__map[key]
            link_prev, link_next, _ = link
            link_prev[1] = link_next
            link_next[0] = link_prev
            link[0] = root
            link[1] = first
            root[1] = first[0] = link
        else:
            root[1] = first[0] = self._OrderedDict__map[key] = [root, first, key]
            dict_setitem(self, key, value)

演示：

>>> d = MyOrderedDict([('a', '1'), ('b', '2')])
>>> d
MyOrderedDict([('a', '1'), ('b', '2')])
>>> d.prepend('c', 100)
>>> d
MyOrderedDict([('c', 100), ('a', '1'), ('b', '2')])
>>> d.prepend('a', d['a'])
>>> d
MyOrderedDict([('a', '1'), ('c', 100), ('b', '2')])
>>> d.prepend('d', 200)
>>> d
MyOrderedDict([('d', 200), ('a', '1'), ('c', 100), ('b', '2')])

操纵OrderedDict对象的独立功能

该函数通过接受dict对象，键和值来做同样的事情。我个人更喜欢上课：

from collections import OrderedDict

def ordered_dict_prepend(dct, key, value, dict_setitem=dict.__setitem__):
    root = dct._OrderedDict__root
    first = root[1]

    if key in dct:
        link = dct._OrderedDict__map[key]
        link_prev, link_next, _ = link
        link_prev[1] = link_next
        link_next[0] = link_prev
        link[0] = root
        link[1] = first
        root[1] = first[0] = link
    else:
        root[1] = first[0] = dct._OrderedDict__map[key] = [root, first, key]
        dict_setitem(dct, key, value)

演示：

>>> d = OrderedDict([('a', '1'), ('b', '2')])
>>> ordered_dict_prepend(d, 'c', 100)
>>> d
OrderedDict([('c', 100), ('a', '1'), ('b', '2')])
>>> ordered_dict_prepend(d, 'a', d['a'])
>>> d
OrderedDict([('a', '1'), ('c', 100), ('b', '2')])
>>> ordered_dict_prepend(d, 'd', 500)
>>> d
OrderedDict([('d', 500), ('a', '1'), ('c', 100), ('b', '2')])

使用OrderedDict.move_to_end()（Python> = 3.2）

Python
3.2引入
了该OrderedDict.move_to_end()方法。使用它，我们可以在O（1）时间将现有键移动到字典的任一端。

>>> d1 = OrderedDict([('a', '1'), ('b', '2')])
>>> d1.update({'c':'3'})
>>> d1.move_to_end('c', last=False)
>>> d1
OrderedDict([('c', '3'), ('a', '1'), ('b', '2')])

如果只需一步就可以插入一个元素并将其移到顶部，则可以直接使用它来创建prepend()包装器（此处未显示）。

创建一个新的OrderedDict-慢！！！

如果您不想这样做并且 性能不是问题， 那么最简单的方法是创建一个新的dict：

from itertools import chain, ifilterfalse
from collections import OrderedDict


def unique_everseen(iterable, key=None):
    "List unique elements, preserving order. Remember all elements ever seen."
    # unique_everseen('AAAABBBCCDAABBB') --> A B C D
    # unique_everseen('ABBCcAD', str.lower) --> A B C D
    seen = set()
    seen_add = seen.add
    if key is None:
        for element in ifilterfalse(seen.__contains__, iterable):
            seen_add(element)
            yield element
    else:
        for element in iterable:
            k = key(element)
            if k not in seen:
                seen_add(k)
                yield element

d1 = OrderedDict([('a', '1'), ('b', '2'),('c', 4)])
d2 = OrderedDict([('c', 3), ('e', 5)])   #dict containing items to be added at the front
new_dic = OrderedDict((k, d2.get(k, d1.get(k))) for k in \
                                           unique_everseen(chain(d2, d1)))
print new_dic

输出：

OrderedDict([('c', 3), ('e', 5), ('a', '1'), ('b', '2')])