問題描述

問題：我需要多次使用同一個游標。然后發(fā)現(xiàn)一個for循環(huán)這個游標就沒有用了。然后我使用了

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

但是a for循環(huán)之后b,c都不能用了。之后又想到用深度拷貝：

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

這樣就可以使用。但是這樣會不會增加內(nèi)存的使用～！

請問怎么使用才是最優(yōu)美的呢？謝謝

問題解答

可以使用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’}]