xlwt 是一個(gè)用于處理 Excel 文件的 Python 庫(kù),但它不支持直接創(chuàng)建和編輯圖表
pip install xlwt
pip install openpyxl
import xlwt
# 創(chuàng)建一個(gè)新的 Excel 工作簿
workbook = xlwt.Workbook()
# 添加一個(gè)工作表
worksheet = workbook.add_sheet("Sheet1")
# 寫入數(shù)據(jù)
for i in range(10):
worksheet.write(i, 0, i)
worksheet.write(i, 1, i * 2)
# 保存文件
workbook.save("example.xls")
import openpyxl
from openpyxl.chart import BarChart, Reference
# 加載已有的 Excel 文件
workbook = openpyxl.load_workbook("example.xls")
worksheet = workbook.active
# 創(chuàng)建一個(gè)條形圖
chart = BarChart()
chart.title = "Bar Chart"
chart.x_axis.title = "X Axis"
chart.y_axis.title = "Y Axis"
# 設(shè)置圖表數(shù)據(jù)
data = Reference(worksheet, min_col=1, max_col=2, min_row=1, max_row=10)
categories = Reference(worksheet, min_col=0, max_col=0, min_row=1, max_row=10)
chart.add_data(data, titles_from_data=True)
chart.set_categories(categories)
# 將圖表添加到工作表
worksheet.add_chart(chart, "D1")
# 保存文件
workbook.save("example_with_chart.xlsx")
這個(gè)示例首先使用 xlwt 創(chuàng)建一個(gè)包含數(shù)據(jù)的 Excel 文件,然后使用 openpyxl 向該文件中添加一個(gè)簡(jiǎn)單的條形圖。請(qǐng)注意,openpyxl 只能處理 .xlsx 格式的文件,因此在這個(gè)示例中,我們將文件保存為 .xlsx 格式。