Python获取数据库数据并保存在excel表格中的方法

下面我将为您提供“Python获取数据库数据并保存在excel表格中的方法”的完整实例教程。

1. 环境准备

在进行代码编写前，您需要确保具备以下环境：

• Python 3.x环境
• MySQL数据库
• MySQL Python包（可以使用pip安装）

2. 导入必要的库

在开始编写代码前，你需要导入以下库：

import pymysql  # MySQL Python包
import xlwt     # Excel数据写入库

3. 定义数据库连接函数

定义一个函数用于连接MySQL数据库：

def connect_mysql():
    '''
    连接MySQL数据库
    '''
    host = 'localhost'   # 主机名
    name = 'test'        # 数据库名
    user = 'root'        # 用户名
    password = 'password'       # 密码

    # 连接MySQL数据库
    try:
        conn = pymysql.connect(host=host, port=3306, user=user, passwd=password, db=name, charset='utf8')
        print("数据库连接成功")
        return conn
    except Exception as e:
        print("数据库连接失败：", e)

其中，host、name、user和password依次为数据库主机名、数据库名、用户名以及密码。端口默认为3306，charset默认为'utf8'。

4. 获取数据库数据

定义一个函数获取MySQL数据库中的数据，并返回数据：

def get_data():
    '''
    获取MySQL数据库数据
    '''
    conn = connect_mysql()
    cursor = conn.cursor()

    # 查询语句
    sql = """
        SELECT * FROM student
    """

    try:
        cursor.execute(sql)
        result = cursor.fetchall()  # 获取所有数据
        return result
    except Exception as e:
        print("获取数据库数据失败：", e)

其中，SQL语句可以根据实际情况进行修改。

5. 保存数据到Excel表格中

定义一个函数将获取到的数据保存到Excel表格中：

def store_to_excel(data):
    '''
    将数据保存到Excel表格中
    '''
    # 创建Excel表格
    wb = xlwt.Workbook(encoding='utf-8')
    sheet = wb.add_sheet('sheet1')

    # 设置表头
    sheet.write(0, 0, '编号')
    sheet.write(0, 1, '姓名')
    sheet.write(0, 2, '年龄')
    sheet.write(0, 3, '性别')
    sheet.write(0, 4, '科目')

    # 逐条写入数据
    for i in range(len(data)):
        sheet.write(i+1, 0, data[i][0])  # 编号
        sheet.write(i+1, 1, data[i][1])  # 姓名
        sheet.write(i+1, 2, data[i][2])  # 年龄
        sheet.write(i+1, 3, data[i][3])  # 性别
        sheet.write(i+1, 4, data[i][4])  # 科目

    # 保存Excel表格
    wb.save('data.xls')

在该函数中，首先创建了一个Excel表格，然后定义了表头，最后逐条向表格中写入数据，并保存表格。

6. 示例演示

最后提供两个示例说明：

示例1

data = get_data()        # 获取数据
store_to_excel(data)     # 将数据保存到Excel表格中

运行上述代码后，程序会从数据库中获取数据，并将数据保存到一个名为data.xls的Excel表格中。

示例2

如果要获取特定条件下的数据，可以修改SQL语句，例如：

sql = """
    SELECT * FROM student WHERE age > 20
"""

表示获取年龄大于20岁的学生信息。

总结

通过以上6步，我们就可以实现将MySQL数据库中的数据获取并保存至Excel表格的功能。当然，如果您对Excel表格的处理细节上还有其他需求，需要自己进行设置。