ヒートマップ 軸ラベルを中央に移動

Question

コード
import matplotlib.pyplot as plt
import numpy as np
import seaborn as sns


X, Y = np.mgrid[:5, :5]
C = np.array(
    [[2.9,2.1,1.4,1.5,4.7],
     [0.69,0.24,2.0,1.0,-0.74],
     [1.5,0.55,-2.0,-4.4,-4.2],
     [-0.56,0.94,-1.6,-0.7,-0.63]]
)
#print(f"X: shape={X.shape}", X, sep="\n")
#print(f"Y: shape={Y.shape}", Y, sep="\n")
#print(f"C: shape={C.shape}", C, sep="\n")

fig, ax = plt.subplots(figsize=(20, 10))
ax.pcolormesh(X, Y, C,edgecolors="w",alpha=0.5)

sns.heatmap(C, annot=True,cmap='coolwarm', linewidths=.5)
sns.set(font_scale=2)
ax.set_xticks([0, 1, 2, 3, 4])
ax.set_xticklabels(['A', 'B', 'C', 'D', 'E'],fontsize='small')


ax.set_yticks([0, 1, 2, 3, 4])
ax.set_yticklabels(['0', '200', '400', '600', '800'],rotation=45,fontsize='small')
 
ax.set_title('figure')
ax.set_xlabel('area')
ax.set_ylabel('distance')

plt.show()

以上のコードから表示できたのは下記に示しています。

このx軸のABCDEを各メッシュの中央に移動したいのですが、どのようにすればよいか分かりません。
ただ、x軸のみを変えたいと考えています。y軸は中央に寄せないつもりです。
ご教授ください。

Answer 1

【追記】
ax.set_xticks([0, 1, 2, 3, 4])をコメントアウトしたところ、他は質問のコードのままでX軸のラベルは中央表示になりました。

---

ax.set_xticks([0, 1, 2, 3, 4], 0.5)とすると良いかと思います。

【追記】
上記はこちらの環境ではDeprecationWarningが出ましたが、グラフは表示されました。

Comments

[yuuu39]

ありがとうございます。
移動できました。