利用Python打造一个Excel记账模板

import openpyxl
from openpyxl.styles import PatternFill, Font
from openpyxl.chart import PieChart, BarChart, Reference

# 创建工作簿并设置工作表
wb = openpyxl.Workbook()
ws = wb.active
ws.title = "记账数据"

# 设置表头
headers = ["日期", "收入", "支出", "预算", "备注", "预算百分比", "预算状态"]
ws.append(headers)

# 设置列宽
ws.column_dimensions['A'].width = 15
ws.column_dimensions['B'].width = 15
ws.column_dimensions['C'].width = 15
ws.column_dimensions['D'].width = 15
ws.column_dimensions['E'].width = 20
ws.column_dimensions['F'].width = 15
ws.column_dimensions['G'].width = 15

# 设置数据格式
for row in ws.iter_rows(min_row=2, max_row=100, min_col=1, max_col=7):
    for cell in row:
        if cell.column == 1:
            cell.number_format = 'YYYY-MM-DD'
        elif cell.column in [2, 3, 4]:
            cell.number_format = '#,##0.00'
        elif cell.column == 6:
            cell.number_format = '0.00%'

# 添加示例数据
example_data = [
    ["2025-01-01", 5000, 3000, 4000, "工资", "=IF(SUM(D:D)=0, \"N/A\", SUM($C$1:C2)/SUM(D:D))", "=IF(F2>1, \"超支\", \"正常\")"],
    ["2025-01-02", 0, 1500, 2000, "购物", "=IF(SUM(D:D)=0, \"N/A\", SUM($C$1:C3)/SUM(D:D))", "=IF(F3>1, \"超支\", \"正常\")"],
    ["2025-01-03", 2000, 1000, 1500, "兼职", "=IF(SUM(D:D)=0, \"N/A\", SUM($C$1:C4)/SUM(D:D))", "=IF(F4>1, \"超支\", \"正常\")"],
]
for data in example_data:
    ws.append(data)

# 创建收入支出比例饼图
pie_chart = PieChart()
labels = Reference(ws, min_col=1, min_row=2, max_row=4)
data = Reference(ws, min_col=2, min_row=1, max_row=4)
pie_chart.add_data(data, titles_from_data=True)
pie_chart.set_categories(labels)
pie_chart.title = "收支比例"
ws.add_chart(pie_chart, "I2")

# 创建每日收支对比柱状图
bar_chart = BarChart()
labels = Reference(ws, min_col=1, min_row=2, max_row=4)
data = Reference(ws, min_col=2, min_row=1, max_col=3, max_row=4)
bar_chart.add_data(data, titles_from_data=True)
bar_chart.set_categories(labels)
bar_chart.title = "每日收支对比"
ws.add_chart(bar_chart, "I18")

# 保存Excel文件
wb.save("Excel记账模板.xlsx")
print("文件 Excel记账模板.xlsx 已成功保存。")