python 单独加上图例colormap

python 只需要跑出一个colormap，这里是jet格式的,也可以换成其他的格式都行。

import numpy as np
import matplotlib.pyplot as plt

gradient = np.linspace(0, 1, 150)
gradient = np.vstack((gradient, gradient))

figh = 0.2
fig, axs = plt.subplots(nrows=1, figsize=(3.0, figh))

ax = axs.imshow(gradient, aspect=‘auto’, cmap=‘jet’)

plt.axis(‘off’)
plt.savefig(‘legend.jpg’)
plt.show()

下面是结果