メッシュ図に数値を入力する

Question

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


X, Y = np.mgrid[:5, :6]
C = np.array(
    [[0, 1, 2, 3, 4],
     [5, 6, 7, 8, 9],
     [10, 11, 12, 13, 14],
     [15, 16, 17, 18, 19]]
)
#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=(5, 5))
ax.pcolormesh(X, Y, C,edgecolors="w",alpha=0.5)

plt.show()

表示された図です。この中に数値を入力したいのですが可能でしょうか？

イメージはこのような図です。
よろしければ、ご教授ください。

Comments

[jbpb0]

import seaborn as sns
sns.heatmap(C, annot=True)
plt.show()

ではダメでしょうか?

[yuuu39]

返信が遅れてしまい申し訳ございません。
上記のコードで表示することができました。
ありがとうございます。

Answer 1

Axes.text() で各セルにテキストとして値を描画すればいいのではないでしょうか。

matplotlib.axes.Axes.text — Matplotlib 3.1.2 documentation

import numpy as np
from matplotlib import pyplot as plt

X, Y = np.mgrid[:5, :6]
C = np.array(
    [[0, 1, 2, 3, 4], [5, 6, 7, 8, 9], [10, 11, 12, 13, 14], [15, 16, 17, 18, 19]]
)

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

for i in range(C.shape[0]):
    for j in range(C.shape[1]):
        ax.text(i + 0.5, j + 0.5, C[i, j], ha="center", va="center", color="k")
plt.show()

Comments

[yuuu39]

ありがとうございます。
この方法でも、表示することができました。