如何使用Python函数来连接数据库并执行查询操作？

连接数据库并执行查询操作是Python中使用数据库的常见操作。Python提供了多个库用于连接和操作不同类型的数据库，例如MySQL、SQLite、PostgreSQL等。下面是一个使用Python函数连接数据库并执行查询操作的示例：

1. 首先，你需要安装所需的数据库驱动。例如，如果你要连接MySQL数据库，可以使用pymysql库。可以通过在终端运行以下命令来安装它：

pip install pymysql

2. 导入pymysql库：

import pymysql

3. 使用pymysql.connect()函数连接到数据库。该函数需要传入数据库的连接参数，例如主机名、用户名、密码和数据库名称：

def connect_to_database():
    connection = pymysql.connect(
        host='localhost',
        user='username',
        password='password',
        database='database_name'
    )
    return connection

4. 创建一个游标对象，用于执行SQL语句：

def execute_query(connection, query):
    cursor = connection.cursor()
    cursor.execute(query)
    return cursor

5. 执行查询操作，并获取结果：

def fetch_data(cursor):
    result = cursor.fetchall()
    return result

6. 关闭数据库连接：

def close_connection(connection):
    connection.close()

7. 完整的示例代码如下所示：

import pymysql

def connect_to_database():
    connection = pymysql.connect(
        host='localhost',
        user='username',
        password='password',
        database='database_name'
    )
    return connection

def execute_query(connection, query):
    cursor = connection.cursor()
    cursor.execute(query)
    return cursor

def fetch_data(cursor):
    result = cursor.fetchall()
    return result

def close_connection(connection):
    connection.close()

# 示例查询
query = "SELECT * FROM table_name"

# 连接数据库
connection = connect_to_database()

# 执行查询操作
cursor = execute_query(connection, query)

# 获取查询结果
result = fetch_data(cursor)
print(result)

# 关闭数据库连接
close_connection(connection)

这是一个基本的示例，展示了如何使用Python函数连接数据库并执行查询操作。根据不同的数据库类型和库，可能需要稍作修改，但整体的连接和查询过程基本相同。