python – 为什么散点图中点的颜色与相应图例中点的颜色不匹配？

内容导读

互联网集市收集整理的这篇技术教程文章主要介绍了python – 为什么散点图中点的颜色与相应图例中点的颜色不匹配？，小编现在分享给大家，供广大互联网技能从业者学习和参考。文章包含2495字，纯文字阅读大概需要4分钟。

内容图文

我通过matplotlib通过下面的代码有一个示例散点图.

import numpy as np
import matplotlib.pyplot as plt

x = np.linspace(0, 100, 501)
y = np.sin(x)

label = 'xy data sample'

plt.scatter(x, y, cmap='plasma', c=x, label=label)
legend_dict = dict(ncol=1, loc='best', scatterpoints=4, fancybox=True, shadow=True)
plt.legend(**legend_dict)
plt.show()

运行上面的代码会生成下面的图表.

已成功绘制色彩图,但图例显示的是全部为蓝色的点,而不是与所选色彩图对应的颜色中的点.为什么会这样？

我尝试在legend_dict中输入cmap =’plasma’,但是会导致下面的错误.

File "/Users/.../
site-packages/matplotlib/axes/_axes.py", line 550, in legend
    self.legend_ = mlegend.Legend(self, handles, labels, **kwargs)
TypeError: __init__() got an unexpected keyword argument 'cmap'

编辑：

我想要的输出是通过所选的色彩图使图例中表示的四个点成为不同的颜色.理想情况下,此示例中的cmap =’plasma’可以使用类似于蓝点的东西产生图例,然后是紫色点,然后是橙红色点,然后是黄色点.虽然彩色条可以作为一种可能的选择,但我还没有查看有关彩条的任何文档.

解决方法:

可以通过plt.colorbar()实现颜色条.这将允许直接看到与颜色对应的值.

让图例中的点显示不同的颜色当然也很好,虽然它不允许提供任何定量信息.

不幸的是,matplotlib没有提供任何内置的方法来实现这一点.因此,一种方法是子类化用于创建图例句柄的图例处理程序并实现此功能.

在这里,我们使用自定义create_collection方法创建一个ScatterHandler,我们在其中创建所需的PathCollection,并通过在图例的legend_map字典中指定它来使用它.

handler_map={ type(sc) : ScatterHandler()}

下面的代码乍一看似乎有点复杂,但是您可能只是复制类而不完全理解它并在代码中使用它.

import numpy as np
import matplotlib.pyplot as plt
from matplotlib.legend_handler import HandlerRegularPolyCollection

class ScatterHandler(HandlerRegularPolyCollection):
    def update_prop(self, legend_handle, orig_handle, legend):
        legend._set_artist_props(legend_handle)
        legend_handle.set_clip_box(None)
        legend_handle.set_clip_path(None)

    def create_collection(self, orig_handle, sizes, offsets, transOffset):
        p = type(orig_handle)([orig_handle.get_paths()[0]],
                              sizes=sizes, offsets=offsets,
                              transOffset=transOffset,
                              cmap=orig_handle.get_cmap(),
                              norm=orig_handle.norm )

        a = orig_handle.get_array()
        if type(a) != type(None):
            p.set_array(np.linspace(a.min(),a.max(),len(offsets)))
        else:
            self._update_prop(p, orig_handle)
        return p


x = np.linspace(0, 100, 501)
y = np.sin(x)*np.cos(x/50.)

sc = plt.scatter(x, y, cmap='plasma', c=x, label='xy data sample')

legend_dict = dict(ncol=1, loc='best', scatterpoints=4, fancybox=True, shadow=True)
plt.legend(handler_map={type(sc) : ScatterHandler()}, **legend_dict)

plt.show()

内容总结

以上是互联网集市为您收集整理的python – 为什么散点图中点的颜色与相应图例中点的颜色不匹配？全部内容，希望文章能够帮你解决python – 为什么散点图中点的颜色与相应图例中点的颜色不匹配？所遇到的程序开发问题。 如果觉得互联网集市技术教程内容还不错，欢迎将互联网集市网站推荐给程序员好友。

内容备注

版权声明：本文内容由互联网用户自发贡献，该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务，不拥有所有权，不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容， 请发送邮件至 gblab@vip.qq.com 举报，一经查实，本站将立刻删除。

内容手机端