面试题：Python操作MongoDB数据库之数据插入

Question

使用Python的pymongo库，如何向MongoDB数据库中的一个集合（collection）插入一条文档（document）以及多条文档？请给出示例代码，并解释每条关键代码的作用。

Answer 1

以下是使用pymongo库向MongoDB数据库集合插入单条和多条文档的示例代码及解释：

1. 插入单条文档

import pymongo

# 连接MongoDB数据库
client = pymongo.MongoClient("mongodb://localhost:27017/")

# 选择数据库
db = client["your_database_name"]

# 选择集合
collection = db["your_collection_name"]

# 定义要插入的文档
document = {
    "name": "John",
    "age": 30,
    "city": "New York"
}

# 插入单条文档
result = collection.insert_one(document)

# 打印插入文档的_id
print("Inserted document ID:", result.inserted_id)

• import pymongo：导入pymongo库，用于操作MongoDB。
• client = pymongo.MongoClient("mongodb://localhost:27017/")：创建一个MongoClient对象，连接到本地运行的MongoDB服务器，默认端口是27017。
• db = client["your_database_name"]：选择要使用的数据库，将your_database_name替换为实际的数据库名。
• collection = db["your_collection_name"]：选择要操作的集合，将your_collection_name替换为实际的集合名。
• document = {... }：定义要插入的文档，这是一个Python字典，键值对对应文档的字段和值。
• result = collection.insert_one(document)：使用insert_one方法插入单条文档，返回一个InsertOneResult对象。
• print("Inserted document ID:", result.inserted_id)：打印插入文档的唯一标识符_id。

2. 插入多条文档

import pymongo

# 连接MongoDB数据库
client = pymongo.MongoClient("mongodb://localhost:27017/")

# 选择数据库
db = client["your_database_name"]

# 选择集合
collection = db["your_collection_name"]

# 定义要插入的多条文档
documents = [
    {
        "name": "Alice",
        "age": 25,
        "city": "San Francisco"
    },
    {
        "name": "Bob",
        "age": 35,
        "city": "Los Angeles"
    }
]

# 插入多条文档
result = collection.insert_many(documents)

# 打印插入文档的_id列表
print("Inserted document IDs:", result.inserted_ids)

• documents = [...]：定义一个包含多个文档的列表，每个文档也是一个Python字典。
• result = collection.insert_many(documents)：使用insert_many方法插入多条文档，返回一个InsertManyResult对象。
• print("Inserted document IDs:", result.inserted_ids)：打印插入的多个文档的_id列表。