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’}]