尝试手动处理：

import matplotlib as mpl
import numpy as np

fig, ax = plt.subplots(figsize=(1, 4))
# vmin, vmax =  np.nanmin(A), np.nanmax(A)
cmap = mpl.colormaps['rocket']
norm = mpl.colors.Normalize(0, 1)  # or vmin, vmax
cbar = fig.colorbar(mpl.cm.ScalarMappable(norm, cmap), ax)
plt.tight_layout()
plt.show()

输出：

据我所知，仅绘制颜色条是不可能的。要查看颜色条的整个光谱，您可以使用范围从0到100的示例热图：

import seaborn as sns
import matplotlib.pyplot as plt

A = [[a] for a in range(100)] # sample heatmap

fig, ax = plt.subplots()

sns.heatmap(A, cbar=False, ax=ax)

cax = fig.add_axes([1.0, 0.0, 0.0, 1.0])  # Adjust the position and size

plt.show()