Question

我试图用 matplotlib 来绘制值, 并有几类不同的点可以绘制。例如 :

Number   class     x     y 
1        wood      0.2   9.4
3        glass     7.6   4.6

我如何能代表不同班级在一个有不同颜色的地块上优雅地代表这些不同的班级呢?

Answer 1

如果您有不同种类的数据,您可以使用 class 。

import matplotlib.pyplot as plt

class DataSet(object):
    def __init__(self, name, x_data, y_data, size, color):
        self.name = name
        self.x_data = x_data
        self.y_data = y_data
        self.size = size
        self.color = color

wood = DataSet( wood , [1,2,3,4], [5,7,4,1], 300,  red )
glass = DataSet( glass , [0,2,0,4], [2,7,4,1], 100,  blue )
paper = DataSet( paper , [1,7,3,6], [9,9,1,9], 500,  #a5a5a5 )
plastic = DataSet( plastic , [3,2,5,5], [1,5,1,8], 700,  yellow )

categories = [wood, glass, paper, plastic]
#Here we plot all the data:
for cat in categories:
    plt.scatter(cat.x_data, cat.y_data, s=cat.size , color=cat.color, label=cat.name)

plt.legend(loc= lower right , scatterpoints=1)
plt.show()

这给了我们:

""https://i.sstatic.net/pQC07.png" alt="此处输入图像描述"/ >

高贵高贵! 优雅高贵!

友情链接