高效使用 PyMongo 進行 MongoDB 查詢和插入操作

插入到集合中：

要將記錄（在MongoDB中稱為文檔）插入到集合中，使用insert_one()方法。insert_one()方法的第一個參數是一個包含文檔中每個欄位的名稱和值的字典。

import pymongo

myclient = pymongo.MongoClient("mongodb://localhost:27017/")
mydb = myclient["mydatabase"]
mycol = mydb["customers"]

mydict = { "name": "John", "address": "Highway 37" }

x = mycol.insert_one(mydict)

返回 _id 欄位：

insert_one()方法返回一個InsertOneResult對象，該對象具有一個屬性inserted_id，其中包含插入的文檔的ID。

mydict = { "name": "Peter", "address": "Lowstreet 27" }

x = mycol.insert_one(mydict)

print(x.inserted_id)

插入多個文檔：

要在MongoDB中的集合中插入多個文檔，使用insert_many()方法。insert_many()方法的第一個參數是包含要插入的數據的字典列表。

mylist = [
  { "name": "Amy", "address": "Apple st 652"},
  { "name": "Hannah", "address": "Mountain 21"},
  { "name": "Michael", "address": "Valley 345"},
  # ...
]

x = mycol.insert_many(mylist)

# 列印插入文檔的 _id 值列表
print(x.inserted_ids)

插入多個文檔，指定ID：

如果不希望MongoDB為文檔分配唯一的ID，可以在插入文檔時指定 _id 欄位。確保這些值是唯一的，兩個文檔不能具有相同的 _id。

mylist = [
  { "_id": 1, "name": "John", "address": "Highway 37"},
  { "_id": 2, "name": "Peter", "address": "Lowstreet 27"},
  { "_id": 3, "name": "Amy", "address": "Apple st 652"},
  # ...
]

x = mycol.insert_many(mylist)

# 列印插入文檔的 _id 值列表
print(x.inserted_ids)

查找一個：

要從MongoDB集合中選擇數據，可以使用find_one()方法，它返回選擇中的第一個文檔。

import pymongo

myclient = pymongo.MongoClient("mongodb://localhost:27017/")
mydb = myclient["mydatabase"]
mycol = mydb["customers"]

x = mycol.find_one()

print(x)

查找所有：

要從MongoDB集合中選擇所有文檔，可以使用find()方法。在find()方法中不指定參數會給出與SQL中的SELECT *相同的結果。

import pymongo

myclient = pymongo.MongoClient("mongodb://localhost:27017/")
mydb = myclient["mydatabase"]
mycol = mydb["customers"]

for x in mycol.find():
  print(x)

僅返回部分欄位：

如果想僅返回部分欄位，可以在find()方法的第二個參數中指定要包含在結果中的欄位。

import pymongo

myclient = pymongo.MongoClient("mongodb://localhost:27017/")
mydb = myclient["mydatabase"]
mycol = mydb["customers"]

for x in mycol.find({}, {"_id": 0, "name": 1, "address": 1}):
  print(x)

不允許在同一對象中同時指定0和1值：

不允許在同一對象中同時指定0和1值，除非其中一個欄位是 _id 欄位。

import pymongo

myclient = pymongo.MongoClient("mongodb://localhost:27017/")
mydb = myclient["mydatabase"]
mycol = mydb["customers"]

# 會出現錯誤，因為同時指定了0和1值
for x in mycol.find({}, {"name": 1, "address": 0}):
  print(x)

高級查詢：

要進行高級查詢，可以在查詢對象中使用修飾符作為值。例如，使用大於修飾符 $gt 可以查找 "address" 欄位以字母 "S" 或更高字母（按字母順序）開頭的文檔。

import pymongo

myclient = pymongo.MongoClient("mongodb://localhost:27017/")
mydb = myclient["mydatabase"]
mycol = mydb["customers"]

myquery = { "address": { "$gt": "S" } }

mydoc = mycol.find(myquery)

for x in mydoc:
  print(x)

使用正則表達式過濾：

您還可以使用正則表達式作為修飾符，但正則表達式只能用於查詢字元串。

例如，要僅查找 "address" 欄位以字母 "S" 開頭的文檔，可以使用正則表達式 {"$regex": "^S"}。

import pymongo

myclient = pymongo.MongoClient("mongodb://localhost:27017/")
mydb = myclient["mydatabase"]
mycol = mydb["customers"]

myquery = { "address": { "$regex": "^S" } }

mydoc = mycol.find(myquery)

for x in mydoc:
  print(x)

最後

為了方便其他設備和平臺的小伙伴觀看往期文章：公眾號搜索Let us Coding，或者掃描下方二維碼，關註公眾號，即可獲取最新文章。

看完如果覺得有幫助，歡迎點贊、收藏和關註