使用Python操作MongoDB

38 篇文章 7 订阅
11 篇文章 0 订阅

        大家好,在当今数字化时代,数据的管理和处理变得越来越重要。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。

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

寒秋丶

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值