Matplotlib绘制拆线图自定义曲线颜色显示
1、在Matplotlib绘制拆线图或散点图时,利用颜色映射可以根据不同的数值显示曲线的颜色。
import matplotlib as mpl import matplotlib.pyplot as plt import numpy as np # 生成示例数据 x = np.linspace(0, 10, 50) y = np.sin(x) fig, ax = plt.subplots(figsize=(10, 6), layout='constrained') cmap = (mpl.colors.ListedColormap( ['cyan', 'blue', 'green','red'] ).with_extremes(under='yellow', over='magenta')) # 计算每个点对应的颜色值 norm = plt.Normalize(min(y), max(y)) colors = [cmap(norm(value)) for value in y] # 绘制折线图并设置颜色 for i in range(len(x) - 1): plt.plot([x[i], x[i + 1]], [y[i], y[i + 1]], 'o-', color=colors[i]) # 设置图表标题和坐标轴标签 plt.title("Sine Wave with Color Mapping") plt.xlabel("X") plt.ylabel("Y") plt.show()2、条形图posx and posy should be finite values报错问题:可能是条形图数值列存在空值,用0或数值填充后可解决。
