python-3.x - 如何使用 Matplotlib 自定义图形

Question

我需要帮助来使用 Matplotlib 自定义图形。
我想画这样的图表。

我的python代码是：

import numpy as np
import matplotlib as mpl
import matplotlib.pyplot as plt

data=np.loadtxt('input.txt', delimiter = ',')

x = np.arange(0,64)
y = np.arange(0,64)

fig, ax = plt.subplots()
im = ax.imshow(data)

#customize colorbar
cmap = mpl.colors.ListedColormap(['royalblue','blue','red'])    

im = ax.contourf(y,x,data,cmap=cmap)
fig.colorbar(im)

plt.show()

我的输出是：

所以我该怎么做 ？
谢谢你。

标签： python-3.xmatplotlib