Python数据可视化之在Excel中插入图表

from openpyxl import load_workbook
from openpyxl.drawing.image import Image
import matplotlib.pyplot as plt
import pandas as pd

def insert_sheet_to_chart_bar(filename, df):
    # 提取数据  [{'A会员数量': 1783, 'B会员数量': 648}]
    categories = list(df.columns)
    values = list(df.values.flatten().tolist())
    # 生成柱状图
    plt.figure(figsize=(8, 4))
    bars = plt.bar(categories, values, color=['#1f77b4', '#ff7f0e'])

    # 设置中文显示
    plt.rcParams['font.sans-serif'] = ['SimHei']
    plt.title("会员数量对比")
    plt.ylabel("数量")

    # 添加数值标签
    for bar in bars:
        height = bar.get_height()
        plt.text(bar.get_x() + bar.get_width() / 2., height,
                 f'{int(height)}',
                 ha='center', va='bottom')

    # 保存图表图片
    chart_path = 'membership_chart.png'
    plt.savefig(chart_path, dpi=300, bbox_inches='tight')
    plt.close()

    # 创建Excel并插入图表
    wb = Workbook()
    ws = wb.active

    # 插入图表图片
    img = Image(chart_path)
    img.width = 1000  # 调整宽度
    img.height = 550  # 调整高度
    ws.add_image(img, 'D2')  # 插入到 D2 单元格位置
    wb.save(filename)

from openpyxl import load_workbook
from openpyxl.drawing.image import Image
import matplotlib.pyplot as plt
import pandas as pd

def insert_sheet_to_chart_bar(filename, df, sheet_name):
    """
    将柱状图插入到指定的 Excel 工作表中
    
    参数:
    filename -- Excel 文件路径
    df -- 包含图表数据的 pandas DataFrame 对象
    sheet_name -- 目标工作表名称
    """
    
    # 提取数据：从 DataFrame 中获取列名和数据值
    categories = list(df.columns)  # 获取列名作为柱状图分类标签
    values = list(df.values.flatten().tolist())  # 将数据展平并转换为列表形式
    
    # 生成柱状图
    plt.figure(figsize=(8, 4))
    bars = plt.bar(categories, values, color=['#1f77b4', '#ff7f0e'])
    
    # 设置中文显示：指定字体以确保中文标题和标签显示正确
    plt.rcParams['font.sans-serif'] = ['SimHei']
    plt.title("会员数量对比")  # 设置图表标题
    plt.ylabel("数量")  # 设置纵轴标签
    
    # 添加数值标签：在每个柱子上方显示具体数值
    for bar in bars:
        height = bar.get_height()
        plt.text(bar.get_x() + bar.get_width() / 2., height,
                 f'{int(height)}',
                 ha='center', va='bottom')
    
    # 保存图表图片：将生成的图表保存为 PNG 格式
    chart_path = 'membership_chart.png'
    plt.savefig(chart_path, dpi=300, bbox_inches='tight')
    plt.close()
    
    # 读取 Excel 文件：加载已存在的 Excel 文件以便进行修改
    wb = load_workbook(filename)
    
    # 获取指定工作表：确保图表插入到正确的工作表中
    if sheet_name in wb.sheetnames:
        ws = wb[sheet_name]
    else:
        ws = wb.create_sheet(sheet_name)  # 如果工作表不存在，则创建
    
    # 插入图表图片：加载图表图片并调整大小，然后插入到指定位置
    img = Image(chart_path)
    img.width = 1000  # 设置图片宽度
    img.height = 550  # 设置图片高度
    ws.add_image(img, 'D2')  # 将图片插入到 D2 单元格位置
    
    # 保存修改后的 Excel 文件：确保所有更改被保存
    wb.save(filename)

# 导入 pandas 库
import pandas as pd

# 创建示例 DataFrame
data = {'月份': ['1月', '2月', '3月'], '销售额': [2500, 3200, 2800]}
df = pd.DataFrame(data)

# 直接读取指定sheet下的数据

# 指定要读取的工作表名称
sheet_name = 'Sheet1'
df = pd.read_excel('data.xlsx', sheet_name=sheet_name)
print(df)

# 调用函数
insert_sheet_to_chart_bar('data.xlsx', df, 'Sheet1')