代码:
import numpy as np
from matplotlib import pyplot as plt
from matplotlib.colors import ListedColormap
y = np.zeros((16384))
y[4096:8193], y[8192:12289], y[12288:] = 1, 2, 3
y = y.reshape(128,128)
color = np.array([[154, 169, 88],
[79, 89, 53],
[182, 141, 68],
[131, 170, 180]])/255 # 输入值是0-1,所以除255
newcmap = ListedColormap(color)
view = plt.imshow(y.astype(int), cmap=newcmap)
bar = plt.colorbar()
bar.set_ticks(np.linspace(0, 16, 17))
bar.set_ticklabels(('Alfalfa', 'Corn-notill', 'Corn-mintill', 'Corn'))
plt.savefig('./image.png',dpi=600)
plt.show()
也可以使用16进制的编码,如下自定义颜色:即 #d62728
plt.plot(x, y, label="xxx", color="#d62728")