如何在 python3 mongodb 查询中从 for 循环返回所有值

Question

我有一个查询 mongodb 集合的函数

def checkDb():
    signals =  mydb['collection']
    findInMongo =  signals.find({}, {'_id': False})
    for x in findInMongo:
        return x

我想返回集合中的所有内容而不将其打印到控制台。当我运行上面的代码时，我得到一个结果：

{'Date': testvalue, 'Open': testvalue, 'High': testvalue, 'Low': testvalue, 'Close': testvalue, 'Volume':testvalue}

当我跑步时

def checkDb():
    signals =  mydb['collection']
    findInMongo =  signals.find({}, {'_id': False})
    for x in findInMongo:
        print(x)

我从我的收藏中获取每一个文档，这就是我想要的。 如何让代码返回每个文档而不是使用print？

Answer 1

问题是您使用的是return，而您可能需要yield：

def checkDb():
    signals =  mydb['collection']
    findInMongo =  signals.find({}, {'_id': False})
    for x in findInMongo:
        return x # This will exit after the first result

return 退出函数，它从函数中返回一个值，退出作用域。另一方面，yield 将继续生成值，直到迭代器耗尽（或findInMongo 在for 循环中没有更多元素）。

相反，做

def checkDb():
    signals =  mydb['collection']
    findInMongo =  signals.find({}, {'_id': False})
    for x in findInMongo:
        yield x

# which allows you to do
vals = list(checkDb())

因为checkDb() 现在是一个迭代器（或者更具体地说，是一个生成器）。之后的python3版本也引入了好看的yield from语法

def checkDb():
    signals =  mydb['collection']
    findInMongo =  signals.find({}, {'_id': False})
    yield from findInMongo # yield from a collection directly

或者，如果这是一个迭代器，你可以只返回 findInMongo

def checkDb():
    signals =  mydb['collection']
    findInMongo =  signals.find({}, {'_id': False})
    return findInMongo # return the iterator directly

所有这些都将支持list(checkDb()) 或[x for x in checkDb()] 语法