Python连接MySQL数据库|极客教程

Python连接MySQL数据库

在Python中，我们可以使用各种库来连接MySQL数据库，执行SQL语句并操作数据库中的数据。本文将详细介绍如何使用Python连接MySQL数据库。

安装MySQL Connector库

要在Python中连接MySQL数据库，我们首先需要安装MySQL Connector库。我们可以使用pip工具来安装MySQL Connector库，命令如下：

pip install mysql-connector-python

连接MySQL数据库

连接MySQL数据库之前，我们需要先获取数据库的连接信息，包括主机名、用户名、密码、数据库名等。假设我们要连接的数据库信息如下：

主机名：localhost
用户名：root
密码：password
数据库名：test

我们可以使用以下代码来连接MySQL数据库：

import mysql.connector

# 连接MySQL数据库
def connect_mysql():
    db = mysql.connector.connect(
        host="localhost",
        user="root",
        passwd="password",
        database="test"
    )
    return db

# 测试连接
db = connect_mysql()
if db.is_connected():
    print("Connected to MySQL database")
else:
    print("Failed to connect to MySQL database")

运行以上代码后，输出应该为Connected to MySQL database，表示成功连接到MySQL数据库。

执行SQL语句

连接到MySQL数据库之后，我们可以执行SQL语句对数据库进行操作，比如创建表、插入数据、查询数据等。以下是一些常用的SQL操作示例：

创建表

# 创建表
def create_table():
    db = connect_mysql()
    cursor = db.cursor()
    cursor.execute("CREATE TABLE customers (id INT AUTO_INCREMENT PRIMARY KEY, name VARCHAR(255), email VARCHAR(255))")
    print("Table created successfully")

# 测试创建表
create_table()

插入数据

# 插入数据
def insert_data(name, email):
    db = connect_mysql()
    cursor = db.cursor()
    sql = "INSERT INTO customers (name, email) VALUES (%s, %s)"
    val = (name, email)
    cursor.execute(sql, val)
    db.commit()
    print("Data inserted successfully")

# 测试插入数据
insert_data('Alice', 'alice@example.com')

查询数据

# 查询数据
def select_data():
    db = connect_mysql()
    cursor = db.cursor()
    cursor.execute("SELECT * FROM customers")
    result = cursor.fetchall()
    for row in result:
        print(row)

# 测试查询数据
select_data()

上述示例代码分别演示了如何创建表、插入数据和查询数据。在执行这些代码之前，确保数据库中没有同名的表或数据，否则可能会导致错误。

关闭数据库连接

在操作完成后，我们应该关闭数据库连接以释放资源。以下是关闭数据库连接的示例代码：

# 关闭数据库连接
db.close()
print("Database connection closed")

总结

本文介绍了如何在Python中连接MySQL数据库，并执行一些常用的数据库操作。

Python连接MySQL数据库