python - pymongo中游标的使用

问题描述

问题：我需要多次使用同一个游标。然后发现一个for循环这个游标就没有用了。然后我使用了

a = db.base.find()c = b = a

但是a for循环之后b,c都不能用了。之后又想到用深度拷贝：

import copya = db.base.find()b = copy.deepcopy(a)c = copy.deepcopy(a)

这样就可以使用。但是这样会不会增加内存的使用～！

请问怎么使用才是最优美的呢？谢谢

问题解答

可以使用itertools的tee

In [20]: from itertools import teeIn [21]: x1,x2 = tee(db.x.find())In [22]: list(x1)Out[22]: [{u’_id’: ObjectId(’590026b521d7dd4a1beb3c1a’), u’name’: u’bar’}, {u’_id’: ObjectId(’590026b921d7dd4a1beb3c1b’), u’name’: u’foo’}]In [23]: list(x2)Out[23]: [{u’_id’: ObjectId(’590026b521d7dd4a1beb3c1a’), u’name’: u’bar’}, {u’_id’: ObjectId(’590026b921d7dd4a1beb3c1b’), u’name’: u’foo’}]