#coding=utf-8
import tkinter
from tkinter import filedialog
from openpyxl import load_workbook
from datetime import datetime
import matplotlib
import matplotlib.pyplot as plt
import matplotlib.dates as mdates
import time,os
print('使用说明:\n1.尽量不使用原文件\n2.只能绘制xlsx格式的文件,若为xls文件,将其内容复制到新建的xlsx文件中打开\n3.默认第一列为横轴数据,第二列为纵轴数据,每列第一个数据为坐标轴名称')
polt_title=input('【*】请输入作图标题:\n')
name_make=input('【*】请输入绘图人姓名:\n')
name_checker=input('【*】请输入审核人姓名:\n')
print('【*】请选择xlsx表格文件:\n')
time.sleep(0.2)
#get file path
root = tkinter.Tk()
root.withdraw() #hide the root window
try:
xls_path = filedialog.askopenfilename()
except:
pass
root.destroy() #close tk,otherwise the program can't close in the last
#open file and save data
wb=load_workbook(xls_path,read_only=True)
#sheet=wb.get_active_sheet()
#sheets=wb.get_sheet_names()
sheet=wb.worksheets[0]
title=sheet.title
temp=[]
data=[]
for c in range(1+sheet.max_column):
for r in range(1+sheet.max_row):
if sheet.cell(row=r+1,column=c+1).value==None:
continue
temp.append(sheet.cell(row=r+1,column=c+1).value)
data.append(temp)
temp=[]
wb.close()
xlabel=data[0][0].encode('utf-8').decode('utf-8')
xdata=[datetime.strptime(str(date),'%Y%m%d')for date in data[0][1:]]
ylabel=data[1][0]
ydata=data[1][1:]
print('当前横坐标:'+xlabel)
print('当前纵坐标:'+ylabel)
#plot picture
fig=plt.figure(num=title,figsize=(14,7),dpi=80) #获取图对象
ax=fig.add_subplot(111) #获取坐标轴对象
ax.xaxis.set_major_formatter(mdates.DateFormatter('%Y%m')) #坐标显示格式
ax.xaxis.set_major_locator(mdates.MonthLocator()) #坐标间隔
ax.set_title(polt_title+'\n',fontsize=23)
matplotlib.rcParams['font.sans-serif']=['SimHei'] #指定默认字体,解决编码问题
plt.plot(xdata,ydata,color='blue',linewidth=1.0,linestyle="-",)
plt.xlabel(xlabel)
plt.ylabel(ylabel)
plt.subplots_adjust(left=0.05,right=0.95,top=0.9,bottom=0.1) #设置图占的位置,按总量为1的比例来算
plt.title('绘图人:'+name_make+' '+'审核人:'+name_checker,size=15,loc = 'right')
plt.savefig(title)
plt.show()
path=os.getcwd()
print('图像已保存至:'+path+'\\'+title+'.png'+'(2s后关闭窗口)')
time.sleep(2)
由xlsx文件绘折线图(使用openpyxl,matplotlib,tkinter...)
最新推荐文章于 2022-09-29 15:38:49 发布