ヒートマップのy軸を調整したい

Question

実現したいこと

ヒートマップのy軸を調整したい

結果

該当のソースコード

import pandas as pd
import seaborn as sns
import matplotlib.pyplot as plt
import numpy as np
import plotly.express as px
sns.set()
path=r"C:\test\heat-a.xlsx"
df=pd.read_excel(path)
print(df)
yticks=["0 0.1 0.2 0.3 0.4 0.5 0.6 0.7 0.8 0.9 1.0"]
sns.heatmap(df,yticklabels=yticks)
plt.xlabel('episode')
plt.ylabel('class')

plt.show()

データの中身

  1話        2話        3話       4話        5話      6話        7話        8話   10話       11話       12話       13話
0  0.008130  0.000000  0.045455  0.00000  0.000000  0.0000  0.000000  0.000000  0.00  0.000000  0.000000  0.000000
1  0.016260  0.000000  0.000000  0.00000  0.000000  0.0000  0.000000  0.000000  0.00  0.000000  0.000000  0.000000
2  0.000000  0.000000  0.000000  0.02222  0.000000  0.0000  0.000000  0.000000  0.00  0.000000  0.000000  0.000000
3  0.016260  0.000000  0.045455  0.00000  0.010638  0.0250  0.000000  0.015873  0.00  0.000000  0.039735  0.014925
4  0.048780  0.000000  0.000000  0.04444  0.031915  0.0125  0.028571  0.000000  0.00  0.023256  0.006623  0.037313
5  0.219512  0.130435  0.227270  0.11111  0.053191  0.1250  0.042857  0.070365  0.25  0.069767  0.086093  0.097015
6  0.357714  0.565217  0.431818  0.35556  0.425532  0.3375  0.342857  0.460317  0.25  0.279070  0.370861  0.417910
7  0.227642  0.260870  0.181818  0.31111  0.361702  0.4375  0.442857  0.317460  0.50  0.418605  0.337748  0.343284
8  0.089431  0.021739  0.204545  0.13333  0.074468  0.0500  0.085714  0.063492  0.00  0.093023  0.125828  0.067164
9  0.016260  0.021739  0.068182  0.02222  0.425530  0.0125  0.057143  0.063492  0.00  0.116279  0.033113  0.022388

やりたいこと．質問

y軸を0 0.1 0.2 0.3 0.4 0.5 0.6 0.7 0.8 0.9 1.0の目盛りにしたいです
y軸がおかしな位置になっているので，これを正しい位置にするにはどうすればいいですか？

Answer 1

下のようなイメージでしょうか

ヒートマップのticksは中央に打たれるので、0.5の位置に0、1.5の位置に0.1, 2.5の位置に 0.2, .... となっています。
なので、ticksのラベルだけではなく、ticksをうつ位置も指定しなおす必要があります。今回は位置はnp.arange()で作ればいいでしょう。
ax.set_yticks(np.arange(0, len(yticks)), labels=yticks)

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

sns.set()
#df = pd.read_excel(path)

ax = sns.heatmap(df)

yticks = ['0', '0.1', '0.2', '0.3', '0.4', '0.5', '0.6', '0.7', '0.8', '0.9', '1.0']
ax.set_yticks(np.arange(0, len(yticks)), labels=yticks, rotation=0)
ax.set_xlabel('episode')
ax.set_ylabel('class')

plt.show()

Answer 2

yticks=["0 0.1 0.2 0.3 0.4 0.5 0.6 0.7 0.8 0.9 1.0"]
sns.heatmap(df,yticklabels=yticks)

↓ 変更

heat_map = sns.heatmap(df)
heat_map.set_yticks(np.arange(0, df.shape[0], 0.1*df.shape[0]).astype('int'))
heat_map.set_yticklabels((np.arange(0, df.shape[0], 0.1*df.shape[0])/df.shape[0]).round(1))

で、一番下の「1.0」以外は表示されると思います