python(pandas, matplot)　DBからグラフに書き出す　その２

f:id:game-allergy:20210331143049p:plain

前回matplotでグラフに書き出しをしたが、１つのグラフを出すより、２つのグラフを並べて比較する…ってのが現実的な使い方だと思う。
ということで、当日の天気と、翌日の天気情報を比較できるグラフを作成してみる。

◆グラフを２つ表示する

#===============================
# DB内テーブル中身の確認（pandas）
# matplotでグラフ書き出し
# 複数グラフ作成
#===============================
"""
ＤＢのテーブルには７２時間分（72個）のデータが並んでいる。
当日、翌日、翌々日のうち、
「当日、翌日」を比較するグラフを作成する
"""

import sqlite3
import pandas as pd
import matplotlib.pyplot as plt

#-----------------------------
# DB接続、カーソル起動
#-----------------------------
dbname = 'TEST.db'
connection = sqlite3.connect(dbname)
cur = connection.cursor()
#-----------------------------
# データを２種類、DBから引き出す
#-----------------------------
# 0-24時までのデータ(data1)
sql1 = 'SELECT temp, hour FROM WeatherTable LIMIT 24'
c1 =cur.execute(sql1)
result1 = c1.fetchall()
#print(result1)

# 次の日の0-24時までのデータ(data2)
sql2 = 'SELECT temp, hour FROM WeatherTable LIMIT 24,24'
c2 = cur.execute(sql2)
result2 = c2.fetchall()
#print(result2)

#-----------------------------
# DataFrameにする
#-----------------------------
#データフレーム（表に書き出すイメージ）
df1 = pd.DataFrame(result1,columns=["temp", "hour"])
df2 = pd.DataFrame(result2,columns=["temp", "hour"])

#-----------------------------
# グラフの準備（２つのグラフ）
#-----------------------------
# グラフのX軸、Y軸の指定
x1 = df1.hour
y1 = df1.temp
x2 = df2.hour
y2 = df2.temp

# 表示エリアの設定
# fig内にchart1,chart2が横並びする形式にする
# まずは、figのオブジェクト生成
fig = plt.figure()

# グラフ上限値を決めておく（5℃～25℃）
tempMin = 5
tempMax = 25

# (data1)----------------------------
# add_subplot(行、列、場所)
chart1 = fig.add_subplot(1,2,1)
# データプロット
chart1.plot(x1, y1)
# Y軸の最小、最大値を取得
#ymin, ymax = plt.ylim()
plt.ylim(tempMin, tempMax)
# グラフ内のラベル表記
chart1.set_title("Today")
chart1.set_xlabel("Time")
chart1.set_ylabel("Temperature(℃)")
# (data2)----------------------------
# add_subplot(行、列、場所)
chart2 = fig.add_subplot(1,2,2)
# データプロット
chart2.plot(x2, y2)
# data1で取得したY軸情報をdata2に設定（同じにする）
#plt.ylim(ymin, ymax)
plt.ylim(tempMin, tempMax)
# グラフ内のラベル表記
chart2.set_title("Tomorrow")
chart2.set_xlabel("Time")
chart2.set_ylabel("Temperature(℃)")

#-----------------------------
# グラフの表示
#-----------------------------
fig.tight_layout() 
plt.show()

#==================
# ディスプレイにデータ表示
#==================
# ディスプレイ表示省略しない
pd.set_option("display.max_colwidth", None)
pd.set_option("display.max_rows", None)

# ディスプレイに表示
print(df1)
print(df2)
#==================
#DB終了処理
#==================
cur.close()
connection.close()

print('--------------------')
print('end of line')
input()

スクリプト概要

取得した情報が７２時間分＝７２個のデータで、これをどうやって当日、翌日にわけようかと…そうしないと２つのグラフにならないし。とりあえずSQLで出す時点で条件分けて２つのデータを抽出することにしてみた。その後、各データfetchall()で全読みしたものをリストに。そのリストをデータフレームにしてグラフ化しやすい状態に。データフレームのオブジェクトにした後はｘ軸にhour、y軸にtempと割り当てればグラフ化できる、こりゃ簡単だ。

ちょっとハマったのはグラフを２つにする箇所。グラフ２つを１つの表示エリアに出す…には、plt.figure()でオブジェクトを作成しておくところ、ここがキモだ。これが出来上がれば後はこの中にグラフ２つ、chart1,chart2を設定していけばいい。

２つのグラフの最大値がバラバラになってしまう問題は、plt.ylim(**,**)で直打ちした。。。最初作成したグラフから最小値、最大値を取得して次のグラフに値をセットする方法もあったけど、見たい領域が決まっているなら自分で指定したほうが早い。今の季節なら５℃～２５℃くらいでいいでしょってことで。

◆実行結果

f:id:game-allergy:20210414164131j:plain