matplotlib/seabornでメッシュグリッド表示(ヒートマップ表示)

matplotlib/seabornでメッシュグリッド表示(ヒートマップ表示)

環境

• Python ver : 3.7.7
• matplotlib ver : 3.2.1
• seaborn ver : 0.10.1

matplotlibでメッシュグリッド表示

import numpy as np
import matplotlib.pyplot as plt

x = np.arange(-1,1,0.05)
y = np.arange(-1,1,0.05)
X,Y = np.meshgrid(x,y)
Z = X**2 + Y**2

fig = plt.figure()
aa = plt.pcolormesh(X,Y,Z,cmap='plasma')
fig.colorbar(aa)
plt.xlabel('X')
plt.ylabel('Y')
plt.show()

※補足
cmapの例は以下を参照
https://matplotlib.org/examples/color/colormaps_reference.html

seabornでメッシュグリッド表示

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

x = np.arange(-1,1,0.05)
y = np.arange(-1,1,0.05)
X,Y = np.meshgrid(x,y)
Z = X**2 + Y**2

sns.heatmap(Z,vmin=0.0,vmax=1.5)
plt.show()

seabornでメッシュグリッド表示(アノテーション表示あり)

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

x = np.arange(-1,1,0.2)
y = np.arange(-1,1,0.2)
X,Y = np.meshgrid(x,y)
Z = X**2 + Y**2

sns.heatmap(Z,vmin=0.0,vmax=2.0,annot=True)
plt.show()