Pymongo游标迭代替代方案咨询：生成索引对应作者与书籍数组

嘿，我明白你想要更简洁的方式来生成这两个索引对应的数组，替代原来循环逐个append的写法。这里有几个实用的替代方案供你参考：

这种方式只需要查询一次数据库，把所有文档缓存到列表后，再分别提取字段生成数组，避免重复查询的开销：

# 一次性获取所有文档并缓存
docs = list(collection.find())
# 用列表推导式快速生成两个数组
authors = [doc['author'] for doc in docs]
books = [doc['book'] for doc in docs]

好处是代码简洁直观，而且后续如果还需要用到文档的其他字段，直接用缓存的docs列表即可，不用再查数据库。

这个方法更高效，只需要遍历一次游标，就能同时生成两个数组，省去缓存整个文档列表的内存开销（适合数据量较大的场景）：

cursor = collection.find()
# 用生成器表达式逐个提取字段，再通过zip拆分
authors, books = zip(*((doc['author'], doc['book']) for doc in cursor))
# 如果你需要的是列表而不是元组，转一下类型就行
authors = list(authors)
books = list(books)

原理是生成器会逐个从游标里取出文档，返回(author, book)的元组，zip(*...)会把这些元组按位置拆分，最终得到两个对应索引的序列。

如果你的项目本来就用Pandas处理数据，这种方式会更省心，还能顺便做一些数据清洗或分析：

import pandas as pd

# 把游标数据转成DataFrame
df = pd.DataFrame(list(collection.find()))
# 提取字段并转成列表
authors = df['author'].tolist()
books = df['book'].tolist()

这种方式适合数据量较大、后续需要对数据做更多操作的场景，Pandas的DataFrame能提供很多便捷的数据处理方法。

内容的提问来源于stack exchange，提问作者Sreeragh A R