Pythonでヒートマップを並べる方法について

Question

前提・実現したいこと

seabornで作ったヒートマップをこちらの記事のように並べたいと思います．

元となるcsvデータはこのような「outputi.csv(i=1~10)」というものです．（例）output0.csv

しかしi=0~9の10枚について並べようとしてfor i in range( num_figure):とするとValueError: num must be 1 <= num <= 10, not 0と返されてしまってどうすべきか分かりません．

また元のデータ名を「t=ivxvy.csv(iを0から100おきに10個並べる)」（つまりt=0vxvy.csv，t=100vxvy.csv，t=200vxvy.csv,...みたいな感じです）としても同様のことができるようにしたいです．

次に、10個並べるように実行したときに下の画像のように1つ1つの画像がつぶれてしまいます．

出力のイメージとしてはネットの拾い物ですがこんな感じに全部正方形になってほしいです．

10個のヒートマップの横に「何色がどの程度の数に対応するのか」といったバーもつけたいなと思います。

よろしくお願いします．

該当のソースコード

import seaborn as sns
import pandas as pd
import numpy as np
import matplotlib.pyplot as plt


num_figure =10  

fig = plt.figure(figsize=(10.0, 10.0))
for i in range(1, num_figure+1):  
    df = pd.read_csv('output{}.csv'.format(i),index_col=0) 
    df = df.iloc[20:60, 20:60] 
    ax = fig.add_subplot(2, 5, i)                 
    sns.heatmap(df, cbar=False, cmap='CMRmap_r',xticklabels=3,yticklabels=3)             
    ax.set_title("t={}".format(i))           
    plt.xlabel("vx")
    plt.ylabel("vy")
plt.tight_layout()                       
fig.suptitle('particle density')        
plt.savefig('heatmaps.png')
plt.show()

補足情報（FW/ツールのバージョンなど）

Anacondaは本日最新にしました

Comments

[meg_]

・エラーメッセージは全文掲載してください。
・pythonのバージョン、モジュールのバージョンは
何ですか？

Answer 1

フィギュアを設定して、その中に作画領域サブプロットを定義しましょう
そして、どのサブプロットに作画するかを指定します。

# 基本領域フィギュアの中に作画領域サブプロットを2行2列設定
# 基本領域の大きさは10x8、X軸・Y軸の単位は共有する
fig, axes = plt.subplots(2, 2, figsize=(10, 8), sharex=True, sharey=True)

import matplotlib.pyplot as plt
import seaborn as sns
import pandas as pd

fig, axes = plt.subplots(2, 2, figsize=(10, 8), sharex=True, sharey=True)

data = pd.read_csv('0vxvy.csv', index_col=0)
data = data.iloc[20:60, 20:60] # まわりの範囲を省略

# axに作画するサブプロットを指定する（左上は[0, 0]、右下が[1, 1]になります）
sns.heatmap(data, ax=axes[0, 0], cmap='Oranges');
sns.heatmap(data, ax=axes[0, 1], cmap='Greens');
sns.heatmap(data, ax=axes[1, 0], cmap='Purples');
sns.heatmap(data, ax=axes[1, 1], cmap='Reds');

(疑問)真ん中につく0は何だろう？

Comments

[Kinsho]

こちらを参考にさせていただいたらできました！
ありがとうございます