Python 使用 Pandas 和 XlsxWriter 套装 – 3

简介

Pandas 是 Python 中一个非常流行的数据分析库，可以用于数据清洗、分析以及可视化等方面。而 XlsxWriter 则是一个功能强大的 Python Excel 操作库，可以用于创建精美的 Excel 报表。本教程将介绍如何使用 Pandas 和 XlsxWriter 套装来生成 Excel 报表，并演示一些常见的操作。

安装

pip install pandas
pip install XlsxWriter

生成 Excel 文件

import pandas as pd
import xlsxwriter

# 创建一个新的 Excel 文件
workbook = xlsxwriter.Workbook('example.xlsx')

# 创建一个工作表
worksheet = workbook.add_worksheet()

# 写入数据
data = [
    ['John', 'Smith', 'Male', 24],
    ['Jane', 'Doe', 'Female', 32],
    ['Bob', 'Johnson', 'Male', 41],
    ['Mary', 'Smith', 'Female', 27]
]

for row_num, row_data in enumerate(data):
    for col_num, col_data in enumerate(row_data):
        worksheet.write(row_num, col_num, col_data)

# 关闭文件
workbook.close()

使用 Pandas 读取 Excel 文件

import pandas as pd

# 读取 Excel 文件
df = pd.read_excel('example.xlsx')

print(df)

使用 Pandas 和 XlsxWriter 写入 Excel 文件

import pandas as pd
import xlsxwriter

# 创建一个新的 Excel 文件
workbook = xlsxwriter.Workbook('example.xlsx')

# 创建一个工作表
worksheet = workbook.add_worksheet()

# 写入数据
data = [
    ['John', 'Smith', 'Male', 24],
    ['Jane', 'Doe', 'Female', 32],
    ['Bob', 'Johnson', 'Male', 41],
    ['Mary', 'Smith', 'Female', 27]
]

# 使用 Pandas 将数据写入工作表
df = pd.DataFrame(data, columns=['First Name', 'Last Name', 'Gender', 'Age'])
df.to_excel(worksheet, index=False)

# 关闭文件
workbook.close()

使用 Pandas 和 XlsxWriter 向 Excel 文件添加图表

import pandas as pd
import xlsxwriter

# 创建一个新的 Excel 文件
workbook = xlsxwriter.Workbook('example.xlsx')

# 创建一个工作表
worksheet = workbook.add_worksheet()

# 写入数据
data = [
    ['John', 'Smith', 'Male', 24],
    ['Jane', 'Doe', 'Female', 32],
    ['Bob', 'Johnson', 'Male', 41],
    ['Mary', 'Smith', 'Female', 27]
]

# 使用 Pandas 将数据写入工作表
df = pd.DataFrame(data, columns=['First Name', 'Last Name', 'Gender', 'Age'])
df.to_excel(worksheet, index=False)

# 添加一个图表
chart = workbook.add_chart({'type': 'bar'})

chart.add_series({
    'name': 'Age',
    'categories': '=Sheet1!$A$2:$B$5',
    'values': '=Sheet1!$D$2:$D$5'
})

worksheet.insert_chart('E2', chart)

# 关闭文件
workbook.close()

以上便是使用 Python、Pandas 和 XlsxWriter 套装生成 Excel 报表的主要内容，希望对大家有所帮助。