Matplotlib散点图类

散点图类

#label.py
import matplotlib.pyplot as plt
class Label():
    def __init__(self,
                 x_label='x-label',
                 x_loc='center',
                 x_fontsize=12,
                 y_label='y-label',
                 y_loc='center',
                 y_fontsize=12,
                 title='Title'
                 ):
        self.x_label = x_label
        self.y_label = y_label
        self.x_loc = x_loc
        self.y_loc = y_loc
        self.title = title
        self.x_fontsize = x_fontsize
        self.y_fontsize = y_fontsize

    def label(self):
        plt.title(self.title)
        plt.xlabel(self.x_label,fontsize=self.x_fontsize,loc=self.x_loc)
        plt.ylabel(self.y_label,fontsize=self.y_fontsize,loc=self.y_loc)

#scatter_chart.py
import matplotlib.pyplot as plt
import numpy as np 
from label import Label
class ScatterChart():
    def __init__(self,
                 x=0,
                 y=0,
                 size=15,
                 color='r',
                 label=Label(),
                 ):
        self.x = x
        self.y = y
        self.size = size
        self.color = color
        self.label = label
        
    def show(self):
        plt.scatter(self.x,self.y,s=self.size,c=self.color)
        self.label.label()
        plt.show()

    def example(self):
        """例子展示"""
        x = np.array([1,2,3,4])
        y = np.array([1,4,9,16])
        print('ScatterChart-Example')
        print('需传入ndarray数据')
        print('x = '+str(x))
        print('y = '+str(y))
        print("ScatterChart(x,y,size=15,color='r',label=Label())")
        plt.scatter(x,y,s=self.size,c=self.color)
        self.label.label()
        plt.show()

# a = ScatterChart()
# a.example()