PythonでEXCELグラフを作成する | 棒グラフの作成

# 必要なモジュールをインポート
# openpyxl: EXCELファイルの操作
import openpyxl
from openpyxl.chart import BarChart, Reference
from openpyxl.chart.series import DataPoint


def excel_bar_chart(data, excel_file_path):
    """
    タプルを受け取り、カテゴリの値が20以上の時に赤い棒グラフを描く関数。

    Args:
        data (tuple): 描画するデータを含むタプル。例: (("カテゴリ1", 値1), ("カテゴリ2", 値2), ...)
        excel_file_path (str): 棒グラフを保存するEXCELファイルのパス。
    """
    
    # 新しいEXCELワークブックを作成
    wb = openpyxl.Workbook()
    ws = wb.active
    ws.title = "Chart"

    # データをシートに書き込む
    for row_idx, (category, value) in enumerate(data, start=1):
        ws.cell(row=row_idx, column=1, value=category)
        ws.cell(row=row_idx, column=2, value=value)

    # 棒グラフを作成
    chart = BarChart()
    # chart.type = "bar"  # グラフのタイプを横棒グラフに設定
    chart.title = "月度ごと売上"
    chart.x_axis.title = "月"
    chart.y_axis.title = "値"

    # データ範囲を設定
    data_ref = Reference(ws, min_col=2, min_row=1, max_row=len(data))
    categories_ref = Reference(ws, min_col=1, min_row=1, max_row=len(data))
    chart.add_data(data_ref, titles_from_data=False)
    chart.set_categories(categories_ref)

    # バーの色を条件に応じて設定
    # chart.series[0] はグラフの最初のデータシリーズを取得します
    series = chart.series[0]

    # data を反復処理し、各データポイントにアクセスします
    # enumerate(data) は (index, (category, value)) のタプルを生成します
    # (_, value)とすることで、カテゴリの部分を無視し、値のみを取得します。
    for i, (_, value) in enumerate(data):
        # DataPoint オブジェクトを作成し、現在のインデックスを設定します
        dp = DataPoint(idx=i)
        
        # 値が 20 以上の場合、バーの色を赤に設定します
        if value >= 20:
            # graphicalProperties.solidFill を使ってバーの塗りつぶし色を設定します
            dp.graphicalProperties.solidFill = "FF0000"
        
        # 設定した DataPoint オブジェクトをデータシリーズのデータポイントリストに追加します
        series.dPt.append(dp)


    # グラフをシートに追加
    ws.add_chart(chart, "E5")

    # EXCELファイルを保存
    wb.save(excel_file_path)



    
##############################################################
# 関数のテスト
if __name__ == "__main__":

    # テスト用のタプルデータ
    data = (("4月", 10), ("5月", 20), ("6月", 30), 
            ("7月", 5),  ("8月", 10), ("9月", 30), 
            ("10月", 5), ("11月", 10), ("12月", 20), 
            ("1月", 30),("2月", 30), ("3月", 5))

    # EXCELファイルへのパス
    excel_file_path = "bar_chart.xlsx"

    # 関数の呼び出し
    excel_bar_chart(data, excel_file_path)

pip install openpyxl