matplotlib - 如何为不同的圆圈标签提供颜色图图例

问题描述

我正在用 Matplotlib 绘制不同的圆圈。每个圆圈都有一个标签，每个标签都有一个颜色。我该怎么做才能为这些不同的标签提供颜色图图例？

我在网上尝试了很多解决方案，包括最幼稚的一个，只需添加plt.colorbar()，我会得到错误

RuntimeError('No mappable was found to use for colorbar')

这是我的完整代码。它有点长。请注意，关键部分只是从if labels is None:. 为了完整起见，我只包括所有内容。

def plot_gaussian_circles(loc_list, scale_list, save_path=None, sigma_coe=3, num_to_plot=300, labels=None):
    mu_x_max = -float('inf')
    mu_y_max = -float('inf')

    mu_x_min = float('inf')
    mu_y_min = float('inf')

    color_idx = 0

    rvs = []

    lim_loc_list = loc_list[:num_to_plot]
    lim_scale_list = scale_list[:num_to_plot]
    for a_mu_, a_sigma_ in zip(lim_loc_list, lim_scale_list):
        a_mu = a_mu_.squeeze()
        a_sigma_ = a_sigma_.squeeze()
        if not type(a_sigma_) is np.ndarray:
            a_sigma_ = a_sigma_.numpy()

        radius = sigma_coe * np.max(a_sigma_)
        a_mu_x = a_mu[0]
        a_mu_y = a_mu[1]

        if (a_mu_x + radius) >= mu_x_max:
            mu_x_max = a_mu_x + radius
        if (a_mu_x - radius) <= mu_x_min:
            mu_x_min = a_mu_x - radius

        if (a_mu_y + radius) >= mu_y_max:
            mu_y_max = a_mu_y + radius
        if (a_mu_y - radius) <= mu_y_min:
            mu_y_min = a_mu_y - radius

        if labels is None:
            rv = plt.Circle(a_mu, radius, fill=False, clip_on=False)
        else:
            colors = cm.rainbow(np.linspace(0, 1, len(set(labels))))
            rv = plt.Circle(a_mu, radius, color=colors[labels[color_idx]], fill=False, clip_on=False)

        rvs.append(rv)
        color_idx = (color_idx + 1)

    fig, ax = plt.subplots()
    ax.set_xlabel('X axis')
    ax.set_ylabel('Y axis')

    axes = plt.gca()
    axes.set_xlim([mu_x_min - 1, mu_x_max + 1])
    axes.set_ylim([mu_y_min - 1, mu_y_max + 1])

    for rv in rvs:
        ax.add_artist(rv)

    if not(labels is None):
        # plt.legend(colors, list(range(len(set(labels)))))
        plt.colorbar()
    if save_path is None:
        plt.plot()
        plt.show()
        # plt.savefig('plotcircles_test.png')
    else:
        plt.savefig(save_path, dpi=200)

这里的图像是我目前得到的，而我希望有一个颜色图的图例。

标签： matplotlib

matplotlib - 如何为不同的圆圈标签提供颜色图图例

问题描述

解决方案

推荐阅读