数据转图片存入pdf

数据转图片拼接转pdf

0 用到的包

''' 获取准备预估的数据，没有label值 ''' 
import pandas as pd
import numpy as np
from bs4 import BeautifulSoup
import requests, json
import os
import re
import pymysql
pymysql.install_as_MySQLdb()
from sqlalchemy import create_engine
import warnings
warnings.filterwarnings('ignore')
import pymssql
import matplotlib.pyplot as plt 
import matplotlib.colors as mcolors
from pylab import mpl
mpl.rcParams['font.sans-serif'] = ['SimHei']  # 坐标轴汉字乱码
plt.rcParams['axes.unicode_minus'] = False   #
import random
import math
import time
import seaborn as sns
import datetime 
import gc
from sklearn.preprocessing import LabelEncoder
from sklearn.externals import joblib
import PIL.Image as Image

1 构建文件夹

''' file是要存的图片的地址，file_name是图片类别名称，根据名称构建不同的图片存储文件夹 '''
def mkdir(file_name):
    file = r'D:\Image'
    files = [file +'/' + i   for  i in file_name]
    for i in files:
        if os.path.exists(i):
            pass
        else:
            os.mkdir(i)
''' 清空Image下面素有文件中所有文件，暴力法则先删除后重建 '''            
def rm_file(file_name):
    file = r'D:\Image'
    files = [file +'/' + i   for  i in file_name]
    for f in files:
        shutil.rmtree(f) # 先清空 再重建
        os.mkdir(f)            

2 数据转为图像并保存

def save_first(df2):
       file = r'D:\Image'
	   plt.figure(