利用Python生成Excel炫酷图表

当使用Python进行数据分析时，Excel是一个常见的工具。在Python中，我们可以使用Pandas和Openpyxl等库来读取，操作和生成Excel文件。本教程将介绍如何使用Python和Matplotlib等库在Excel中生成炫酷的图表。

准备工作

在开始之前，我们需要安装一些必要的库：

• Pandas：用于数据处理和读取Excel文件。
• Openpyxl：用于写入Excel文件。
• Matplotlib：用于绘制图表。

安装这些库可以使用pip命令：

pip install pandas openpyxl matplotlib

示例1：生成简单的Excel图表

以下是一个简单的示例，演示如何将一个DataFrame写入Excel文件并在Excel中创建一个饼图：

import pandas as pd
from openpyxl import Workbook
from openpyxl.chart import PieChart, Reference

# 创建DataFrame
data = {
    '语言': ['Python', 'Java', 'C', 'C++', 'JavaScript'],
    '排名': [1, 2, 3, 4, 5],
    '数量': [30, 25, 20, 15, 10]
}
df = pd.DataFrame(data)

# 创建Excel文件
wb = Workbook()
ws = wb.active
ws.title = '数据'

# 将DataFrame写入Excel文件
for r in dataframe_to_rows(df, index=False, header=True):
    ws.append(r)

# 创建饼图
pie = PieChart()
labels = Reference(ws, min_col=1, min_row=2, max_row=6)
data = Reference(ws, min_col=3, min_row=1, max_row=6)
pie.add_data(data=data)
pie.set_categories(labels)
pie.title = '编程语言排名'
ws.add_chart(pie, 'D1')

# 保存Excel文件
wb.save('chart1.xlsx')

在第2到6行创建了数据，创建了一个DataFrame。接下来，在第9到14行创建了一个Excel文件，将DataFrame写入该文件中。最后，在第17到24行创建了一个饼图并添加到Excel文件中。保存Excel文件。

示例2：生成多个Excel图表

以下是一个示例，演示如何将多个图表写入Excel文件：

import pandas as pd
from openpyxl import Workbook
from openpyxl.chart import LineChart, BarChart, Reference

# 创建DataFrame
data = {
    '地区': ['北京', '上海', '深圳'],
    '销售额': [100, 200, 300],
    '利润': [50, 100, 150]
}
df = pd.DataFrame(data)

# 创建Excel文件
wb = Workbook()
ws1 = wb.active
ws1.title = '销售额'
ws2 = wb.create_sheet(title='利润')

# 将DataFrame写入Excel文件
for r in dataframe_to_rows(df, index=False, header=True):
    ws1.append(r)
for r in dataframe_to_rows(df, index=False, header=True):
    ws2.append(r)

# 创建线形图和柱状图
line = LineChart()
line.title = '销售额'
line.y_axis.title = '销售额'
data = Reference(ws1, min_col=2, min_row=1, max_row=4)
line.add_data(data, titles_from_data=True)
ws1.add_chart(line, 'D5')

bar = BarChart()
bar.title = '利润'
bar.y_axis.title = '利润'
data = Reference(ws2, min_col=3, min_row=1, max_row=4)
categories = Reference(ws2, min_col=2, min_row=1, max_row=4)
bar.add_data(data, titles_from_data=True)
bar.set_categories(categories)
ws2.add_chart(bar, 'C5')

# 保存Excel文件
wb.save('chart2.xlsx')

在第2行到8行创建了数据，并创建了一个DataFrame。接下来，在第11到16行创建了一个Excel文件。第18行到21行将DataFrame写入Excel文件中的两个工作表中。最后，创建了一个线形图和一个柱状图，并将它们添加到两个工作表中。最后，保存Excel文件。

以上两个示例演示了如何使用Python和相关库创建Excel图表，将其写入Excel文件，并将它们视觉化。您可以使用这些示例作为起点来创建更多类型的Excel图表。