Matplotlib - 如何使用对数刻度设置线图的颜色条

Posted

tags:

篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了Matplotlib - 如何使用对数刻度设置线图的颜色条相关的知识,希望对你有一定的参考价值。

我在将颜色条添加到对应于幂律的许多线的图中时遇到了问题。

为了创建非图像绘图的颜色条,我添加了一个虚拟绘图(来自这里的答案:Matplotlib - add colorbar to a sequence of line plots)。

对于色条刻度线不符合绘图的颜色。

我已经尝试改变颜色条的标准,我可以对它进行微调,以便对特定情况准确,但我不能这样做。

def plot_loglog_gauss():
    from matplotlib import cm as color_map
    import matplotlib as mpl

    """Creating the data"""
    time_vector = [0, 1, 2, 4, 8, 16, 32, 64, 128, 256]
    amplitudes = [t ** 2 * np.exp(-t * np.power(np.linspace(-0.5, 0.5, 100), 2)) for t in time_vector]

    """Getting the non-zero minimum of the data"""
    data = np.concatenate(amplitudes).ravel()
    data_min = np.min(data[np.nonzero(data)])

    """Creating K-space data"""
    k_vector = np.linspace(0,1,100)

    """Plotting"""
    number_of_plots = len(time_vector)
    color_map_name = 'jet'
    my_map = color_map.get_cmap(color_map_name)
    colors = my_map(np.linspace(0, 1, number_of_plots, endpoint=True))

    # plt.figure()
    # dummy_plot = plt.contourf([[0, 0], [0, 0]], time_vector, cmap=my_map)
    # plt.clf()

    norm = mpl.colors.Normalize(vmin=time_vector[0], vmax=time_vector[-1])
    cmap = mpl.cm.ScalarMappable(norm=norm, cmap=color_map_name)
    cmap.set_array([])


    for i in range(number_of_plots):
        plt.plot(k_vector, amplitudes[i], color=colors[i], label=time_vector[i])

    c = np.arange(1, number_of_plots + 1)
    plt.xlabel('Frequency')
    plt.ylabel('Amplitude')
    plt.yscale('symlog', linthreshy=data_min)
    plt.xscale('log')
    plt.legend(loc=3)

    ticks = time_vector
    plt.colorbar(cmap, ticks=ticks, shrink=1.0, fraction=0.1, pad=0)

    plt.show()

Plot generated

通过与图例进行比较,您会看到刻度值与实际颜色不匹配。例如,128在colormap中以绿色显示,而在图例中以红色显示。

实际结果应该是线性颜色条。在彩条上以固定间隔打勾(对应于不规则的时间间隔......)。当然,正确的颜色为刻度值。

(最终该图包含许多图(len(time_vector)~100),我降低了图的数量来说明并能够显示图例。)

为了澄清,这就是我想要的结果。

Plot wanted

答案

最重要的原则是保持线图和ScalarMappable的颜色同步。这意味着,线的颜色不应取自独立的颜色列表,而应取自相同的颜色图,并使用与要显示的颜色条相同的标准化。

然后一个主要问题是决定如何处理0,这不能成为一种规则化的一部分。以下是使用SymLogNorm假设0到2之间的线性标度和上面的对数标度的变通方法。

import matplotlib as mpl
import matplotlib.pyplot as plt
import numpy as np

"""Creating the data"""
time_vector = [0, 1, 2, 4, 8, 16, 32, 64, 128, 256]
amplitudes = [t ** 2 * np.exp(-t * np.power(np.linspace(-0.5, 0.5, 100), 2)) for t in time_vector]

"""Getting the non-zero minimum of the data"""
data = np.concatenate(amplitudes).ravel()
data_min = np.min(data[np.nonzero(data)])

"""Creating K-space data"""
k_vector = np.linspace(0,1,100)

"""Plotting"""
cmap = plt.cm.get_cmap("jet")
norm = mpl.colors.SymLogNorm(2, vmin=time_vector[0], vmax=time_vector[-1])

sm = mpl.cm.ScalarMappable(norm=norm, cmap=cmap)
sm.set_array([])

for i in range(len(time_vector)):
    plt.plot(k_vector, amplitudes[i], color=cmap(norm(time_vector[i])), label=time_vector[i])

#c = np.arange(1, number_of_plots + 1)
plt.xlabel('Frequency')
plt.ylabel('Amplitude')
plt.yscale('symlog', linthreshy=data_min)
plt.xscale('log')
plt.legend(loc=3)

cbar = plt.colorbar(sm, ticks=time_vector, format=mpl.ticker.ScalarFormatter(), 
                    shrink=1.0, fraction=0.1, pad=0)

plt.show()

enter image description here

以上是关于Matplotlib - 如何使用对数刻度设置线图的颜色条的主要内容,如果未能解决你的问题,请参考以下文章

Matplotlib折线图(x/y轴刻度刻度标签标签旋转)

Matplotlib 对数刻度刻度标签数字格式

Seaborn 线图对数刻度

使用 Seaborn 和 Matplotlib 在热图和线图的共享子图中对齐 x 轴刻度

使用多个连接的箱线图更改 Matplotlib 中的轴刻度

用Python设置matplotlib.plot的坐标轴刻度间隔以及刻度范围