Python数据库操作函数：连接MySQL，MongoDB等数据库，快速查询和处理数据

Python是一种函数式语言，提供了很多操作数据库的函数，可用于连接MySQL、MongoDB等数据库。Python的数据库操作函数非常强大，它能够快速查询和处理数据，使得数据库的操作变得非常简单。本文将介绍Python的数据库操作函数及其应用场景。

一、连接MySQL数据库

连接MySQL数据库可以使用Python的MySQLdb模块。代码如下：

import MySQLdb

# 打开数据库连接
db = MySQLdb.connect("localhost","testuser","test123","TESTDB" )

# 使用cursor()方法获取操作游标 
cursor = db.cursor()

# 使用execute方法执行SQL语句
cursor.execute("SELECT VERSION()")

# 使用fetchone()方法获取一条数据
data = cursor.fetchone()

print "Database version : %s " % data

# 关闭数据库连接
db.close()

上述代码中，我们首先连接MySQL数据库的testuser用户，并且通过TESTDB数据库连接到了这个用户的数据库。然后使用了MySQLdb模块中的cursor()方法创建了一个游标，这个游标是通过与数据库交互来处理数据的。下面使用了cursor的execute()方法执行了一条SQL语句。最后使用fetchone()方法获取SQL语句返回的结果。

二、连接MongoDB数据库

连接MongoDB数据库可以使用Python的pymongo模块。代码如下：

from pymongo import MongoClient

#连接MongoDB服务器
client=MongoClient('localhost',27017)

#获取MongoDB数据库
db=client.test

#获取MongoDB数据集合
collection=db.score

#查询数据
score=collection.find({"name": "John"})

#输出查询结果
for item in score:
    print(item)

上述代码中，我们首先连接到MongoDB服务器，并获取了test数据库。然后获取了数据库的一个集合score，并对其进行数据查询。最后我们使用了循环语句输出了查询结果。

三、快速查询和处理数据

Python的数据库操作函数中，最常用的就是进行数据查询和处理。我们先看一下查询数据的函数。

1. 查询数据

使用Python的函数进行查询数据，或许比写SQL语句更为直观和容易理解。如下是对MySQL数据库的数据进行查询的代码：

import MySQLdb

# 打开数据库连接
db = MySQLdb.connect("localhost","testuser","test123","TESTDB" )

# 使用cursor()方法获取操作游标 
cursor = db.cursor()

# SQL 查询语句
sql = "SELECT * FROM EMPLOYEE \
       WHERE INCOME > %d" % (1000)
try:
   # 执行SQL语句
   cursor.execute(sql)
   # 获取所有记录列表
   results = cursor.fetchall()
   for row in results:
      fname = row[0]
      lname = row[1]
      age = row[2]
      sex = row[3]
      income = row[4]
      # 打印结果
      print "fname=%s,lname=%s,age=%d,sex=%s,income=%d" % \
             (fname, lname, age, sex, income )
except:
   print "Error: unable to fecth data"

# 关闭数据库连接
db.close()

上述代码中，我们首先连接MySQL数据库。然后使用SQL查询语句进行查询操作，并使用fetchall()方法获取查询结果。最后我们遍历了查询结果，将结果以易于理解的方式进行打印输出。

下面是对MongoDB数据库进行查询的代码：

from pymongo import MongoClient

#连接MongoDB服务器
client=MongoClient('localhost',27017)

#获取MongoDB数据库
db=client.test

#获取MongoDB数据集合
collection=db.score

#查询数据
score=collection.find({"name": "John"})

#输出查询结果
for item in score:
    print(item)

上述代码中，我们使用了Python的pymongo模块进行MongoDB数据库的连接。然后获取了数据库的一个集合score，并对其进行数据查询。最后使用了循环语句输出了查询结果。

2. 处理数据

Python的数据库操作函数中，还有许多用于对数据进行处理和修改的函数。我们以MySQL数据库为例，来介绍一下这些数据处理函数的使用方法。

1) 插入数据

Python的MySQLdb模块提供了一个非常方便的函数来插入数据到数据库中。如下是插入数据的代码：

import MySQLdb

# 打开数据库连接
db = MySQLdb.connect("localhost","testuser","test123","TESTDB" )

# 使用cursor()方法获取操作游标 
cursor = db.cursor()

# SQL 插入语句
sql = "INSERT INTO EMPLOYEE(FIRST_NAME, \
       LAST_NAME, AGE, SEX, INCOME) \
       VALUES ('Mac', 'Mohan', 20, 'M', 2000)"

try:
   # 执行sql语句
   cursor.execute(sql)
   # 提交到数据库执行
   db.commit()
except:
   # 发生错误时回滚
   db.rollback()

# 关闭数据库连接
db.close()

上述代码中，我们首先打开了MySQL数据库连接。然后使用SQL插入语句进行数据插入操作，并使用execute()方法执行这条SQL语句。最后使用了提交操作提交插入的数据到数据库中。

2) 更新数据

更新数据时需要使用MySQL的update语句。下面是一个对数据进行更新操作的Python代码：

import MySQLdb

# 打开数据库连接
db = MySQLdb.connect("localhost","testuser","test123","TESTDB" )

# 使用cursor()方法获取操作游标 
cursor = db.cursor()

# SQL 更新语句
sql = "UPDATE EMPLOYEE SET AGE = AGE + 1 WHERE SEX = '%c'" % ('M')

try:
   # 执行SQL语句
   cursor.execute(sql)
   # 提交到数据库执行
   db.commit()
except:
   # 发生错误时回滚
   db.rollback()

# 关闭数据库连接
db.close()

上述代码中，我们首先打开了MySQL数据库连接。然后使用SQL的update语句进行数据更新操作，并使用execute()方法执行这条SQL语句。最后使用了提交操作提交更新的数据到数据库中。

3) 删除数据

删除数据时需要使用MySQL的delete语句。下面是一个对数据进行删除操作的Python代码：

import MySQLdb

# 打开数据库连接
db = MySQLdb.connect("localhost","testuser","test123","TESTDB" )

# 使用cursor()方法获取操作游标 
cursor = db.cursor()

# SQL 删除语句
sql = "DELETE FROM EMPLOYEE WHERE AGE > %d" % (20)

try:
   # 执行SQL语句
   cursor.execute(sql)
   # 提交修改
   db.commit()
except:
   # 发生错误时回滚
   db.rollback()

# 关闭数据库连接
db.close()

上述代码中，我们首先打开了MySQL数据库连接。然后使用SQL的delete语句进行数据删除操作，并使用execute()方法执行这条SQL语句。最后使用了提交操作提交删除的数据到数据库中。

结语

本文主要介绍了Python数据库操作函数的应用，包括连接MySQL、MongoDB等数据库，快速查询和处理数据等。希望读者通过本文的介绍，可以熟练掌握Python的数据操作函数，从而开发出更加强大的应用程序。