python如何使用matplotlib画散点图使不同类别的点有不同的形状和颜色？

最新推荐文章于 2024-03-18 00:36:00 发布

ErgouChen

最新推荐文章于 2024-03-18 00:36:00 发布

阅读量3.4w

点赞数 11

分类专栏： python

本文链接：https://blog.youkuaiyun.com/qq_33251995/article/details/89161863

版权

文章地址：https://www.chzzz.club/post/239.html

我们在使用聚类算法时，希望最后画出的散点图中，不同的类别的点拥有不同的颜色和不同的形状的，但是matplotlib只能让我们使不同类别拥有不同的颜色，而对形状无能为力。下面这段代码就可以实现这个功能：

import numpy as np
import matplotlib.pyplot as plt

def mscatter(x,y,ax=None, m=None, **kw):
    import matplotlib.markers as mmarkers
    if not ax: ax=plt.gca()
    sc = ax.scatter(x,y,**kw)
    if (m is not None) and (len(m)==len(x)):
        paths = []
        for marker in m:
            if isinstance(marker, mmarkers.MarkerStyle):
                marker_obj = marker
            else:
                marker_obj = mmarkers.MarkerStyle(marker)
            path = marker_obj.get_path().transformed(