【python】【Excel】複数のCSVからデータを抽出しExcelにまとめる、かつ散布図も自動作成する方法

Tkinterとopenpyxlの使い方

■はじめに

前回、CSVのデータを抽出しEXcelにまとめ、かつ散布図も作成する方法を紹介しました。

stjun.hatenablog.com

今回は複数のCSVを一気に処理し、かつすべてのシートに散布図を作成する方法を紹介します。

■コード

#複数のCSVファイルをExcelにまとめ、かつ散布図を自動で作成

import tkinter as tk 
import tkinter.filedialog as fl 
import tkinter.messagebox as mb 
import numpy as np 
import time
import openpyxl as px
from openpyxl.chart import ScatterChart,Reference,Series

root=tk.Tk()

#ボタンを押した時の処理
def get(*args):
    
    #新しいExcelの作成
    wb=px.Workbook()
    
    #エクセル内の各シートの名前
    sheet_title=["計算結果1","計算結果2","計算結果3"]
    
    #複数のCSVファイルを選ぶ
    #ユーザー名にお使いのPCの名前を入れてください
    filetype=[("all file","*")]    
    path=fl.askopenfilenames(initialdir="C:/Users/ユーザー/Desktop",filetypes=filetype,title="select file")
    
    # 読み取った複数のCSVファイルを1つのExcelファイルにまとめる
    for i in range(len(path)):
        data_1, data_2,data_3= np.loadtxt(fname=path[i], skiprows=2, unpack=True)

        #Excel内に新しいシートを作成
        new_sheet=wb.create_sheet("計算結果")
        ws = wb["計算結果"]

        #作成したシートの名前を設定
        ws.title=sheet_title[i]

        #Excel一行目にどんな物理量について計算したか記載するためのリスト
        list_target=["物理量A(単位)","物理量B(単位)","物理量C(単位)"]

        #Excelの1行目に上記リストを記載する
        for i in range(len(list_target)):
            ws.cell(row=1,column=i+1,value=list_target[i])

        #Excelの2行目以降にCSVデータを代入する
        for i in range(len(data_1)):
            ws.cell(row=i+2,column=1,value=data_1[i])
            ws.cell(row=i+2,column=2,value=data_2[i])
            ws.cell(row=i+2,column=3,value=data_3[i])
        
        #グラフとして散布図を作成する
        chart =ScatterChart()
        #グラフのタイトル
        chart.title = "散布図のテスト"
        #x軸のタイトル
        chart.x_axis.title = 'x軸'
        #y軸のタイトル
        chart.y_axis.title = 'y軸'

        #x軸とするデータ
        x_values = Reference(ws, min_col=1, min_row=2, max_row=len(data_1)+1)
        for i in range(2, 4):
            #y軸とするデータ
            y_values = Reference(ws, min_col=i, min_row=1, max_row=len(data_1)+1)
            #x軸とy軸の組み合わせを決める
            con = Series(y_values, x_values, title_from_data=True)
            #マーカーの形
            con.marker.symbol = 'circle' 
            #マーカーの塗りつぶし
            con.marker.graphicalProperties.solidFill ="FFFFFF"
            #マーカーの枠線の色
            con.marker.graphicalProperties.line.solidFill = "000000"
            #散布図に系列を追加
            chart.series.append(con)

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

    #今日の日付・時間を取得（ファイル名に入れるため）
    timestr=time.strftime("%Y%m%d-%H%M%S") #年月日時分秒の形式で取得
    
    #Excelデータをデスクトップに保存する(ユーザー名にお使いのPCの名前を入れてください)
    wb.save(r"/Users/ユーザー/Desktop/test"+timestr+".xlsx")
    
    #処理が完了したことをユーザーに知らせる
    mb.showinfo("確認","Excelファイルの作成が完了しました")
    message["text"]="処理が完了しました"


message=tk.Label(root,text="ファイルを選択してください",width=30)
message.grid(row=0,column=0)

button=tk.Button(text="開く",command=get)
button.grid(row=0,column=1)

root.mainloop()