颜色条的设置
如下程序实现了条形图的颜色渐变,类似于矩阵热力图。
import matplotlib.pyplot as plt import matplotlib.colors as mcolors import pandas as pd import numpy as np df = pd.DataFrame({"x" : np.random.rand(10)*10}) # cmap实际上是一个可以将数值映射为颜色的线性函数 cmap = mcolors.LinearSegmentedColormap.from_list("", ["red", "yellow", "green"]) plt.bar(df.index, df["x"], color=cmap(df.x.values/df.x.values.max())) plt.xlabel('Column', labelpad=12) plt.tight_layout() plt.show(
参考:https://www.matplotlib.org.cn/gallery/color/custom_cmap.html