使用 Python 更新 MongoDB 集合

Question

我有一个 MongoDB 文档是这样的：

我有一个文本文件，其中包含一些单词及其情感分数。如果这个词在 testcollection 中作为 "surfaceEnd"，我想更新一些字段。否则，我想插入一个新行。

 for w in words:
            print w
            if  db.testcollection10.find({ 'surfaceEnd': w }) == True:
                posnum = float(get_positive(cols))
                negnum = float(get_negative(cols))
                db.testcollection.update({ 'surfaceEnd': w}, {"$set": { 'posEnd': posnum,'negEnd': negnum,'findEnd' : 1 }})
                i = i + 1
            else:
                cursor = db.collectionNelly.find({ 'surfaceStart': w })

                for document in cursor:
                    relation =  document['rel']
                    word = document['surfaceEnd'].encode('utf-8')
                    posnum = float(get_positive(cols))
                    negnum = float(get_negative(cols))
                    if 'Synonym' in document['rel']:
                         db.testcollection1.insert ({ 'surfaceStart': w,'posStart': posnum, 'negStart': negnum, 'surfaceEnd': word,'posEnd': posnum,'negEnd': negnum, 'rel' : document['rel'], 'findEnd' : 0  })

很遗憾，无法创建测试集合。这段代码有什么问题？

Answer 1

PyMongo 的查找操作returns 一个游标而不是一个布尔值。您想确定返回了多少条记录并采取相应行动。

for w in words:
        print w
        items = db.testcollection10.find({ 'surfaceEnd': w })
        if items.count() > 0:
            # The surfaceEnd entry was found update it.
        else:
            # The surfaceEnd entry was not found, insert a new one.

使用 Python 更新 MongoDB 集合

Update MongoDB collection using Python

python

mongodb

pymongo