pymongo：如何通過名稱作為字符串訪問MongoDB集合？

Question

我想將csv文件寫入mongoDB。

我想將集合的名稱與csv文件的名稱相同。

以下是我嘗試過的代碼。 這會將集合名稱寫為“集合”。

from pymongo import MongoClient
import pandas as pd
import os

#get all csv files from the directory
dir_path = r'C:\Users\USER\Desktop\project'
output_files = [file for file in os.listdir(dir_path) if file.endswith(".csv")]

client = MongoClient()
db = client.senci

for file in output_files:
     collection = file[-4]
     df = pd.read_csv(file) 
     records_ = df.to_dict(orient = 'records')    
     result = db.collection.insert_many(records_ )

Answer 1

您需要使用db.getCollection（） ：

db.get_collection(collection).insert_many(records_ )

從文檔：

返回在功能上等效於使用db.<collectionName>的集合或視圖對象db.<collectionName>