忘れられた備忘録

機械学習の研究をしています.強化学習,逆強化学習が専門です.

テクノロジー

Matplotlibでヒートマップ(カラーバー付き)

matplotlib.pyplot.pcolormeshを使う import numpy as np import matplotlib.pyplot as plt %matplotlib inline x = np.linspace(-4, 4, 200) y = np.linspace(-5, 5, 200) x_mesh, y_mesh = np.meshgrid(x, y) z_mesh = - np.sqrt(x_mesh**2 + y_mesh**2) p…