|
程序功能：已封装的python中的Matplotlib绘图模块，可绘制科技论文中简单的数据可视化。可实现绘制曲线图、柱状图。可实现线条类型、颜色、宽度的设置、可实现对刻度线坐标字体字号设置、可实现图例、网格设置、可实现图中图等其他设置。
# -*- coding: utf-8 -*-
import matplotlib.pyplot as plt
from pylab import *
import numpy as np
from matplotlib.font_manager import FontProperties
#加中文语句
# plt.rcParams['font.sans-serif'] = ['SimHei'] # 解决中文乱码
plt.rcParams['font.sans-serif']=['SimHei'] # 用来正常显示中文标签
plt.rcParams['axes.unicode_minus']=False # 用来正常显示负号
color = ['b','g','r','c','m','y','k','w'] #颜色选择
style = ['-','--','-.',':','.',',','o','v',]#线条类型选择
loc=['upper right','upper left','lower left','lower right','right','center left',
'center right','lower center','upper center','center'] #图例位置选择
'''
titlename:图片标题名称
xlim:设置x坐标轴的范围（可加可不加）
ylim:设置y坐标轴的范围（可加可不加）
xlabel：图片x轴名称
ylabel：图片y轴名称
ticks:设置关键点的记号标签
a,b：绘图的图片的长宽
'''
class Fig:
def __init__(self,a,b):
# self.figure=plt.figure(figsize=(a, b)) #画图中图使用，创建图中图界面并设置图片长和宽
self.figure,self.ax = plt.subplots(figsize=(a,b))
self.show=plt.show()
#绘图
'''
x:输入x的数据代表x轴长度时间
y：输出y的数据是一个列表，如[d,e,f],与输入x个数对应，也是代表三组数据，出三条线
lable：图例的名称 是一个列表，里面是字符串，如["第一个图例"，"第二个图例","第三个图例"]
linewidth：线条宽度
color：线条颜色，是一个列表，里面是字符串。如["r","g","y"]
style:线条类型选择
leged:图例的位置（1，1）这种类型
titlename:图的标题名
xlabel：x轴名称
ylabel：y轴名称
a:是否选择一张图中出多个图，1代表选择，其他代表不选择
b:plt.subplot(b,c,d)函数里的画几个图
c:plt.subplot(b,c,d)画几列
d:plt.subplot(b,c,d)里的第几个
e:是否选择自己设置坐标轴刻度，1代表选择，其他数字不选择
f:x轴坐标轴刻度范围，（a,b）这种形式,从左到右
g:y轴坐标轴刻度范围，（a,b）这种形式,从下到上
font:设置每个图例的字体字号，与输入y对应,为列表数据，可选择font1(英文),font2（中文宋体）
fony:为y轴坐标字体选择,可选择font1(英文),font2（中文宋体）
'''
def fig(self,x,y,lable,linewidth,color,style,leged,titlename,xlabel,ylabel,a,b,c,d,e,f,g,font,fonty):
#判断是否好几个图
if a == 1:
plt.subplot(b,c,d)
#设置字体为，字号
font1 = FontProperties(fname=r"c:\windows\fonts\times.ttf",size=25)#英文Times New Roman字体
font2 = FontProperties(fname=r"c:\windows\fonts\simsun.ttc",size=25)#中文宋体
#为方便选择坐标刻度字体
for i in range(len(font)):
if font[i]==1:
font[i]=font1
else:
font[i]=font2
#判断是否需要自制坐标轴刻度，1代表需要
if e==1:
plt.xlim(f)#设置坐标轴的刻度，范围 例(0,10)
plt.ylim(g)#设置坐标轴的刻度，范围 例(-1，1)
for i in range(len(y)):#绘制y条线
plt.plot(x,y[i],label=lable[i],linestyle=style[i],linewidth=linewidth,color=color[i])
# plt.legend(loc=loc[0],bbox_to_anchor=leged,prop=font[i])
#图例参数解析 bbox_to_anchor=(num1,num2) num1 用于控制 legend 的左右移动，值越大越向右边移动，num2 用于控制 legend 的上下移动，
# 值越大，越向上移动。用于微调图例的位置。
#为方便选择坐标y轴刻度字体设置
if fonty==1:
fonty=font1
else:
fonty=font2
plt.legend(loc=loc[0],bbox_to_anchor=leged,prop=font[0])#加图例 位置信息，字体,字体大小
plt.title(titlename)
plt.xlabel(xlabel,fontproperties=font1,size=30)#图片x轴名称,字体，字号
plt.ylabel(ylabel,fontproperties=fonty,size=35)#图片y轴名称，字体，字号
# 添加网格显示
plt.grid(linestyle="-", alpha=0.8)
plt.tick_params(direction='in',labelsize=25)#坐标字体大小设置 direction='in'刻度线显示在绘图区内测
#如下为设置坐标轴刻度的字体
labels = self.ax.get_xticklabels() + self.ax.get_yticklabels()
[label.set_fontname('Times New Roman') for label in labels]
#绘制图中图
'''
left1,bottom1,width1,height1 分别代表图中图距离y,x轴的距离,图中图本身的长度和宽度
x,y 代表 图中图需要画出的x，y的图形
linewidth,color 图中图中线条的宽度和颜色
xlim,ylim 图中图所要绘制的x,y的大小长高
xlabel,ylabel,title 图中图中横坐标，纵坐标 标题的题目
'''
def image(self,left1,bottom1,width1,height1,x,y,linewidth,color,style,xlim,ylim,xlabel,ylabel,title):
# left1,bottom1,width1,height1=left1,bottom1,width1,height1#图中图位置和宽高
ax2=self.figure.add_axes([left1,bottom1,width1,height1])
for i in range(len(y)):#图中图中绘制y条线
plt.plot(x,y[i],linewidth=linewidth,linestyle=style[i],color=color[i])
plt.grid(linestyle="-", alpha=0.8) #加网格
plt.xlim(xlim)#图中图的横坐标范围
plt.ylim(ylim)#图中图的纵坐标范围
ax2.set_xlabel(xlabel) #设置x轴名字
ax2.set_ylabel(ylabel)#设置y轴名字
ax2.set_title(title)#设置题目
plt.tick_params(direction='in',labelsize=15)#坐标字体大小设置
#如下为设置坐标轴刻度的字体样式
labels = ax2.get_xticklabels() + ax2.get_yticklabels()
[label.set_fontname('Times New Roman') for label in labels]
#绘制散点图
def figur(self,x,y,color,label):#x,y,颜色，图例名称
plt.scatter(x,y,c=color,label=label)#画散点图
plt.legend(loc='upper left', frameon=False)#加图例
#绘制柱状图
'''
x:x轴数据
y:y轴数据
color:柱状图颜色
title：图标题目
ylabel：y轴名称
xlabel：x轴名称
'''
def cylinder(self,left1,bottom1,width1,height1,x,y,color,width,ylim,title,xlabel,ylabel,a,b,c):
ax2=self.figure.add_axes([left1,bottom1,width1,height1])
plt.ylim(ylim)#图中图的纵坐标范围
font1 = FontProperties(fname=r"c:\windows\fonts\times.ttf",size=20)
font2 = FontProperties(fname=r"c:\windows\fonts\simsun.ttc",size=20)#中文宋体
plt.bar(x, y, color = color, width=width,align = 'center')
for x, y in enumerate(y):
plt.text(x, y , '%s' % y, ha='center', va='bottom',fontproperties=font1,size=20)#柱状图上方数字显示
plt.ylim(ylim)#图中图的纵坐标范围
plt.title(title)
plt.ylabel(ylabel)
plt.xlabel(xlabel,fontproperties=font2)
plt.grid(linestyle="-", alpha=0.7)
plt.tick_params(direction='in',labelsize=16)#坐标字体大小设置 direction='in'刻度线显示在绘图区内测
plt.xticks(fontproperties = font1,size = 25) #选择x轴坐标字体设置
labels = ax2.get_xticklabels() + ax2.get_yticklabels()
[label.set_fontname('Times New Roman') for label in labels]
#添加文字,第一个参数是x轴坐标，第二个参数是y轴坐标，以数据的刻度为基准,c表示要加的文字
# plt.text(a,b,c,fontdict={'size':'16.5','color':'k'})
plt.text(a,b,c,fontproperties=font2) #图上文字显示字体字号设置
plt.show()
# #测试代码
# time = np.arange(0, 10,0.01)
# sin = np.sin(time)
# cos = np.cos(time)
# tt=Fig(8,5)
# #demo示例
# tt.fig(time,[sin,cos],['sin(x)','cos(x)'],2,['r','b'],['-','-.'],(1,1),
# " ",'Time(sec)','yaw Angle(rad)',2,3,1,2,1,(0,10),(-1.5, 2.5))
# #绘制图中图
# tt.image(0.3,0.65,0.3,0.2,time,[sin,cos],2,['r','b'],['-','-.'],
# (2,8),(-0.8,0.8),"","","")
# # x=['ff','2','3','4','5','6']
# # y=[81.7315,32.9466,6.2922,86.9391,55.3920,7.9238]
# # tt.cylinder(0.3,0.65,0.3,0.2,x,y,'b',2,(-1,1),'','','')
|