繪圖123

import os
# from matplotlib.pyplot import *
import matplotlib.pyplot as plt
import pandas as pd
# import numpy as np
# import xlsxwriter
from openpyxl.drawing.image import Image
from openpyxl import load_workbookxxxx


os.chdir('C:/Users/xxxx/Desktop/CommitLog/')
output_dir = 'C:/Users/xxxx/Desktop/CommitLog/'
def excel_draw(csv_file, excel_file):
    # pass
    df=pd.read_csv(csv_file,encoding='utf-8',index_col=0 , usecols = [0,1])
    csv_name =csv_file.split('.')[0].split('/')[-1]
    print csv_name
    # print df_x
    # # df = np.array(df_x)
    # x = df_x.values.tolist()
    # print x
    # df_y=pd.read_csv("./Edk2 - Copy.csv",encoding='utf-8', usecols = [0,1])#index_col=0 ,
    # train_data = np.array(df_y)
    # print train_data
    # y = train_data.tolist()
    # print y
    df_plot = df.plot(kind='bar', rot=0, color='green', width=0.35)
    plt.title('%s Comprehensive' % csv_name, fontproperties='SimHei')
    plt.legend(loc=1)
    plt.xlabel('modules', fontproperties='SimHei', fontsize=10)
    plt.xticks(fontproperties='SimHei',fontsize=10,color='black',rotation=30)
    # for a, b in zip(x, y):
    #         plt.text(a, b + 0.05, '%.0f' % b, ha='center', va='bottom', fontsize=17)
    # plt.xlim(0,11)
    plt.ylabel('values', fontproperties='SimHei')
    plt.savefig('./%s.jpg' % csv_name,dpi=100, bbox_inches = 'tight')
    img = Image('./%s.jpg' % csv_name)
    workbook1 = load_workbook(excel_file)
    worksheet = workbook1[csv_name]
    worksheet.add_image(img, 'D2')
    workbook1.save(excel_file)
    # wb = xlsxwriter.Workbook('all1.xlsx')

    # worksheet1.insert_image('D2', './Edk2.jpg')
    plt.show()
csv_file =     os.path.join(output_dir, "Edk2.csv")
print csv_file
excel_file =os.path.join(output_dir, "WW46.xlsx") 
excel_draw(csv_file, excel_file)
# f.figure()
# f.subplot(331)
# bar()

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章