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):
    # 提取數(shù)據(jù)  [{'A會員數(shù)量': 1783, 'B會員數(shù)量': 648}]
    categories = list(df.columns)
    values = list(df.values.flatten().tolist())
    # 生成柱狀圖
    plt.figure(figsize=(8, 4))
    bars = plt.bar(categories, values, color=['#1f77b4', '#ff7f0e'])

    # 設(shè)置中文顯示
    plt.rcParams['font.sans-serif'] = ['SimHei']
    plt.title("會員數(shù)量對比")
    plt.ylabel("數(shù)量")

    # 添加數(shù)值標簽
    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()

    # 創(chuàng)建Excel并插入圖表
    wb = Workbook()
    ws = wb.active

    # 插入圖表圖片
    img = Image(chart_path)
    img.width = 1000  # 調(diào)整寬度
    img.height = 550  # 調(diào)整高度
    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 工作表中
    
    參數(shù):
    filename -- Excel 文件路徑
    df -- 包含圖表數(shù)據(jù)的 pandas DataFrame 對象
    sheet_name -- 目標工作表名稱
    """
    
    # 提取數(shù)據(jù)：從 DataFrame 中獲取列名和數(shù)據(jù)值
    categories = list(df.columns)  # 獲取列名作為柱狀圖分類標簽
    values = list(df.values.flatten().tolist())  # 將數(shù)據(jù)展平并轉(zhuǎn)換為列表形式
    
    # 生成柱狀圖
    plt.figure(figsize=(8, 4))
    bars = plt.bar(categories, values, color=['#1f77b4', '#ff7f0e'])
    
    # 設(shè)置中文顯示：指定字體以確保中文標題和標簽顯示正確
    plt.rcParams['font.sans-serif'] = ['SimHei']
    plt.title("會員數(shù)量對比")  # 設(shè)置圖表標題
    plt.ylabel("數(shù)量")  # 設(shè)置縱軸標簽
    
    # 添加數(shù)值標簽：在每個柱子上方顯示具體數(shù)值
    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)  # 如果工作表不存在，則創(chuàng)建
    
    # 插入圖表圖片：加載圖表圖片并調(diào)整大小，然后插入到指定位置
    img = Image(chart_path)
    img.width = 1000  # 設(shè)置圖片寬度
    img.height = 550  # 設(shè)置圖片高度
    ws.add_image(img, 'D2')  # 將圖片插入到 D2 單元格位置
    
    # 保存修改后的 Excel 文件：確保所有更改被保存
    wb.save(filename)

# 導(dǎo)入 pandas 庫
import pandas as pd

# 創(chuàng)建示例 DataFrame
data = {'月份': ['1月', '2月', '3月'], '銷售額': [2500, 3200, 2800]}
df = pd.DataFrame(data)

# 直接讀取指定sheet下的數(shù)據(jù)

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

# 調(diào)用函數(shù)
insert_sheet_to_chart_bar('data.xlsx', df, 'Sheet1')