大家好,在当今数字化时代,数据的管理和处理变得越来越重要。MongoDB 作为一种强大的 NoSQL 数据库,为我们提供了高效、灵活的数据存储和查询方式。而 Python 作为一种功能强大的编程语言,拥有丰富的库和工具,能够方便地与 MongoDB 进行交互。在本文中,我们将探讨如何使用 Python 来操作 MongoDB 数据库,包括连接数据库、插入数据、查询数据、更新数据和删除数据等操作,希望能给大家带来一些帮助。
一、安装PyMongo
1、通过pip安装
Python 要连接 MongoDB 需要 MongoDB 驱动,这里我们使用 PyMongo 驱动来连接。
python3 -m pip3 install pymongo
也可以指定安装的版本:
python3 -m pip3 install pymongo==3.5.1
2、easy_install 安装
旧版的 Python 可以使用 easy_install 来安装,easy_install 也是 Python 包管理工具。
python -m easy_install pymongo
3、测试pymogo
创建一个测试文件 test_mongodb.py,代码如下:
# -*- coding:utf-8 -*-
import pymongo
执行以上代码文件,如果没有出现错误,表示安装成功。
二、创建数据库
创建数据库需要使用 MongoClient 对象,并且指定连接的 URL 地址和要创建的数据库名。
如下实例中,我们创建的数据库 testdb :
# -*- coding:utf-8 -*-
import pymongo
myclient = pymongo.MongoClient("mongodb://localhost:27017/")
mydb = myclient["testdb"]
在 MongoDB 中,数据库只有在内容插入后才会创建,就是说,数据库创建后要创建集合(数据表)并插入一个文档(记录),数据库才会真正创建。
我们可以读取 MongoDB 中的所有数据库,并判断指定的数据库是否存在:
# -*- coding:utf-8 -*-
import pymongo
myclient = pymongo.MongoClient('mongodb://localhost:27017/')
dblist = myclient.list_database_names()
# dblist = myclient.database_names()
if "testdb" in dblist:
print("数据库已存在!")
database_names 在最新版本的 Python 中已废弃,Python3.7+ 之后的版本改为了 list_database_names()。
三、创建集合
MongoDB 中的集合类似 SQL 的表,MongoDB 使用数据库对象来创建集合,实例如下:
# -*- coding:utf-8 -*-
import pymongo
myclient = pymongo.MongoClient("mongodb://localhost:27017/")
mydb = myclient["testdb"]
mycol = mydb["sites"]
在 MongoDB 中,集合只有在内容插入后才会创建,就是说,创建集合(数据表)后要再插入一个文档(记录),集合才会真正创建。
我们可以读取 MongoDB 数据库中的所有集合,并判断指定的集合是否存在:
# -*- coding:utf-8 -*-
import pymongo
myclient = pymongo.MongoClient('mongodb://localhost:27017/')
mydb = myclient['testdb']
collist = mydb.list_collection_names()
# collist = mydb.collection_names()
if "sites" in collist: # 判断 sites 集合是否存在
print("集合已存在!")
collection_names 在最新版本的 Python 中已废弃,Python3.7+ 之后的版本改为了 list_collection_names()。
四、Python Mongodb 插入文档
MongoDB 中的一个文档类似 SQL 表中的一条记录。
1、插入集合
集合中插入文档使用 insert_one() 方法,该方法的第一参数是字典 name => value 对。
以下实例向 sites 集合中插入文档:
# -*- coding:utf-8 -*-
import pymongo
myclient = pymongo.MongoClient("mongodb://localhost:27017/")
mydb = myclient["testdb"]
mycol = mydb["sites"]
mydict = {"name": "zhangsan", "alexa": "10000", "url": "https://www.baidu.com"}
x = mycol.insert_one(mydict)
print(x)
2、返回 _id 字段
insert_one() 方法返回 InsertOneResult 对象,该对象包含 inserted_id 属性,它是插入文档的 id 值。
# -*- coding:utf-8 -*-
import pymongo
myclient = pymongo.MongoClient('mongodb://localhost:27017/')
mydb = myclient['testdb']
mycol = mydb["sites"]
mydict = {"name": "Google", "alexa": "1", "url": "https://www.google.com"}
x = mycol.insert_one(mydict)
print(x.inserted_id)
如果我们在插入文档时没有指定 _id,MongoDB 会为每个文档添加一个唯一的 id。
3、插入多个文档
集合中插入多个文档使用 insert_many() 方法,该方法的第一参数是字典列表。
# -*- coding:utf-8 -*-
import pymongo
myclient = pymongo.MongoClient("mongodb://localhost:27017/")
mydb = myclient["testdb"]
mycol = mydb["sites"]
mylist = [
{"name": "Taobao", "alexa": "100", "url": "https://www.taobao.com"},
{"name": "QQ", "alexa": "101", "url": "https://www.qq.com"},
{"name": "Facebook", "alexa": "10", "url": "https://www.facebook.com"},
{"name": "知乎", "alexa": "103", "url": "https://www.zhihu.com"},
{"name": "Github", "alexa": "109", "url": "https://www.github.com"}
]
x = mycol.insert_many(mylist)
# 输出插入的所有文档对应的 _id 值
print(x.inserted_ids)
insert_many() 方法返回 InsertManyResult 对象,该对象包含 inserted_ids 属性,该属性保存着所有插入文档的 id 值。
4、插入指定 _id 的多个文档
我们也可以自己指定 id,插入,以下实例我们在 site2 集合中插入数据,_id 为我们指定的:
# -*- coding:utf-8 -*-
import pymongo
myclient = pymongo.MongoClient("mongodb://localhost:27017/")
mydb = myclient["testdb"]
mycol = mydb["site2"]
mylist = [
{"_id": 1, "name": "Google", "address": "Google 搜索"},
{"_id": 2, "name": "Facebook", "address": "脸书"},
{"_id": 3, "name": "Taobao", "address": "淘宝"},
{"_id": 4, "name": "Zhihu", "address": "知乎"}
]
x = mycol.insert_many(mylist)
# 输出插入的所有文档对应的 _id 值
print(x.inserted_ids)
五、Python Mongodb 查询文档
MongoDB 中使用了 find 和 find_one 方法来查询集合中的数据,它类似于 SQL 中的 SELECT 语句。
1、查询一条数据
我们可以使用 find_one() 方法来查询集合中的一条数据。
查询 sites 文档中的第一条数据:
# -*- coding:utf-8 -*-
import pymongo
myclient = pymongo.MongoClient("mongodb://localhost:27017/")
mydb = myclient["testdb"]
mycol = mydb["sites"]
x = mycol.find_one()
print(x)
2、查询集合中所有数据
find() 方法可以查询集合中的所有数据,类似 SQL 中的 SELECT * 操作。
以下实例查找 sites 集合中的所有数据:
# -*- coding:utf-8 -*-
import pymongo
myclient = pymongo.MongoClient("mongodb://localhost:27017/")
mydb = myclient["testdb"]
mycol = mydb["sites"]
for x in mycol.find():
print(x)
3、查询指定字段的数据
我们可以使用 find() 方法来查询指定字段的数据,将要返回的字段对应值设置为 1。
# -*- coding:utf-8 -*-
import pymongo
myclient = pymongo.MongoClient("mongodb://localhost:27017/")
mydb = myclient["testdb"]
mycol = mydb["sites"]
for x in mycol.find({}, {"_id": 0, "name": 1, "alexa": 1}):
print(x)
除了 _id,不能在一个对象中同时指定 0 和 1,如果设置了一个字段为 0,则其他都为 1,反之亦然。
以下实例除了 alexa 字段外,其他都返回:
# -*- coding:utf-8 -*-
import pymongo
myclient = pymongo.MongoClient("mongodb://localhost:27017/")
mydb = myclient["testdb"]
mycol = mydb["sites"]
for x in mycol.find({}, {"alexa": 0}):
print(x)
以下代码同时指定了 0 和 1 则会报错:
# -*- coding:utf-8 -*-
import pymongo
myclient = pymongo.MongoClient("mongodb://localhost:27017/")
mydb = myclient["testdb"]
mycol = mydb["sites"]
for x in mycol.find({}, {"name": 1, "alexa": 0}):
print(x)
报错内容:
pymongo.errors.OperationFailure: Projection cannot have a mix of inclusion and exclusion.
4、根据指定条件查询
我们可以在 find() 中设置参数来过滤数据。
以下实例查找 name 字段为 "RUNOOB" 的数据:
# -*- coding:utf-8 -*-
import pymongo
myclient = pymongo.MongoClient("mongodb://localhost:27017/")
mydb = myclient["testdb"]
mycol = mydb["sites"]
myquery = {"name": "Google"}
mydoc = mycol.find(myquery)
for x in mydoc:
print(x)
5、高级查询
查询的条件语句中,我们还可以使用修饰符。
以下实例用于读取 name 字段中第一个字母 ASCII 值大于 "H" 的数据,大于的修饰符条件为 {"$gt": "H"} :
# -*- coding:utf-8 -*-
import pymongo
myclient = pymongo.MongoClient("mongodb://localhost:27017/")
mydb = myclient["testdb"]
mycol = mydb["sites"]
myquery = {"name": {"$gt": "H"}}
mydoc = mycol.find(myquery)
for x in mydoc:
print(x)
6、使用正则表达式查询
我们还可以使用正则表达式作为修饰符。
正则表达式修饰符只用于搜索字符串的字段。
以下实例用于读取 name 字段中第一个字母为 "R" 的数据,正则表达式修饰符条件为 {"$regex": "^R"} :
# -*- coding:utf-8 -*-
import pymongo
myclient = pymongo.MongoClient("mongodb://localhost:27017/")
mydb = myclient["testdb"]
mycol = mydb["sites"]
myquery = {"name": {"$regex": "^R"}}
mydoc = mycol.find(myquery)
for x in mydoc:
print(x)
7、返回指定条数记录
如果我们要对查询结果设置指定条数的记录可以使用 limit() 方法,该方法只接受一个数字参数。
以下实例返回 3 条文档记录:
# -*- coding:utf-8 -*-
import pymongo
myclient = pymongo.MongoClient("mongodb://localhost:27017/")
mydb = myclient["testdb"]
mycol = mydb["sites"]
myresult = mycol.find().limit(3)
# 输出结果
for x in myresult:
print(x)
8、数据排序
sort() 方法可以指定升序或降序排序。
sort() 方法第一个参数为要排序的字段,第二个字段指定排序规则,1 为升序,-1 为降序,默认为升序。
对字段 alexa 按升序排序:
# -*- coding:utf-8 -*-
import pymongo
myclient = pymongo.MongoClient("mongodb://localhost:27017/")
mydb = myclient["testdb"]
mycol = mydb["sites"]
mydoc = mycol.find().sort("alexa")
for x in mydoc:
print(x)
对字段 alexa 按降序排序:
# -*- coding:utf-8 -*-
import pymongo
myclient = pymongo.MongoClient("mongodb://localhost:27017/")
mydb = myclient["testdb"]
mycol = mydb["sites"]
mydoc = mycol.find().sort("alexa", -1)
for x in mydoc:
print(x)
六、Python Mongodb 修改文档
我们可以在 MongoDB 中使用 update_one() 方法修改文档中的记录。该方法第一个参数为查询的条件,第二个参数为要修改的字段。如果查找到的匹配数据多于一条,则只会修改第一条。
以下实例将 alexa 字段的值 10000 改为 12345:
# -*- coding:utf-8 -*-
import pymongo
myclient = pymongo.MongoClient("mongodb://localhost:27017/")
mydb = myclient["testdb"]
mycol = mydb["sites"]
myquery = {"alexa": "10000"}
newvalues = {"$set": {"alexa": "12345"}}
mycol.update_one(myquery, newvalues)
# 输出修改后的 "sites" 集合
for x in mycol.find():
print(x)
update_one() 方法只能修匹配到的第一条记录,如果要修改所有匹配到的记录,可以使用 update_many()。
以下实例将查找所有以 F 开头的 name 字段,并将匹配到所有记录的 alexa 字段修改为 123:
# -*- coding:utf-8 -*-
import pymongo
myclient = pymongo.MongoClient("mongodb://localhost:27017/")
mydb = myclient["testdb"]
mycol = mydb["sites"]
myquery = {"name": {"$regex": "^F"}}
newvalues = {"$set": {"alexa": "123"}}
x = mycol.update_many(myquery, newvalues)
print(x.modified_count, "文档已修改")
七、Python Mongodb 删除数据
1、删除单个文档
我们可以使用 delete_one() 方法来删除一个文档,该方法第一个参数为查询对象,指定要删除哪些数据。
以下实例删除 name 字段值为 "Taobao" 的文档:
# -*- coding:utf-8 -*-
import pymongo
myclient = pymongo.MongoClient("mongodb://localhost:27017/")
mydb = myclient["testdb"]
mycol = mydb["sites"]
myquery = {"name": "Taobao"}
mycol.delete_one(myquery)
# 删除后输出
for x in mycol.find():
print(x)
2、删除多个文档
我们可以使用 delete_many() 方法来删除多个文档,该方法第一个参数为查询对象,指定要删除哪些数据。
删除所有 name 字段中以 F 开头的文档:
# -*- coding:utf-8 -*-
import pymongo
myclient = pymongo.MongoClient("mongodb://localhost:27017/")
mydb = myclient["testdb"]
mycol = mydb["sites"]
myquery = {"name": {"$regex": "^F"}}
x = mycol.delete_many(myquery)
print(x.deleted_count, "个文档已删除")
3、删除集合中的所有文档
delete_many() 方法如果传入的是一个空的查询对象,则会删除集合中的所有文档:
# -*- coding:utf-8 -*-
import pymongo
myclient = pymongo.MongoClient("mongodb://localhost:27017/")
mydb = myclient["testdb"]
mycol = mydb["sites"]
x = mycol.delete_many({})
print(x.deleted_count, "个文档已删除")
4、删除集合
我们可以使用 drop() 方法来删除一个集合。
以下实例删除了 customers 集合:
# -*- coding:utf-8 -*-
import pymongo
myclient = pymongo.MongoClient("mongodb://localhost:27017/")
mydb = myclient["testdb"]
mycol = mydb["sites"]
mycol.drop()
如果删除成功 drop() 返回 true,如果删除失败(集合不存在)则返回 false。