GenImage.py 文件源码

def generateImage(self, sourceName, fileName, image_kind):
        matplotlib.rcParams['font.family'] = 'SimHei'           #???????linux?????
        matplotlib.rc('font', **self.FONT)               
        mydata = pd.read_csv(self.FILE_PATH + sourceName)       
        mydata.sort_index()

        if image_kind == 'pie':
            mydata.plot(kind='pie', subplots=True, figsize=(10,10), autopct='%1.1f%%', fontsize=20)
        elif image_kind == 'bar':
            mydata.plot(kind='bar', subplots=True, fontsize=10, figsize=(4,6))
        else:
            raise TypeError('????')

        plt.savefig(self.FILE_PATH + 'images/' + fileName,dpi=100)
        plt.close()