Commit 4a0cf3d7 authored by Matthieu Boileau's avatar Matthieu Boileau
Browse files

Add title to colorbar

parent fa883a39
......@@ -109,7 +109,8 @@ class Walk2D:
ax.set_xlim(x.min(), x.max())
ax.set_ylim(y.min(), y.max())
fig.colorbar(lc)
cb = fig.colorbar(lc)
cb.ax.set_title(r"$n_{step}$")
start_color = cmap(t[0])
end_color = cmap(t[-2] * 256 / t.shape[0])
else:
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment