如何快速将字典拆分成多个词典

Question

我找到了一个解决方案，但确实很慢：

def chunks(self,data, SIZE=10000):
    for i in xrange(0, len(data), SIZE):
        yield dict(data.items()[i:i+SIZE])

您有没有使用外部模块（numpy等）的任何想法？

Answer 1

由于字典很大，因此最好将所有涉及的项都保留为迭代器和生成器，像这样

from itertools import islice

def chunks(data, SIZE=10000):
    it = iter(data)
    for i in xrange(0, len(data), SIZE):
        yield {k:data[k] for k in islice(it, SIZE)}

样品运行：

for item in chunks({i:i for i in xrange(10)}, 3):
    print item

输出量

{0: 0, 1: 1, 2: 2}
{3: 3, 4: 4, 5: 5}
{8: 8, 6: 6, 7: 7}
{9: 9}