matplotlib绘图

使用matplotlib可视化数据。图表的基本组成元素包括画布、坐标系、坐标轴、坐标轴标题、图表标题、数据标签、网格线、图例等。

1.建立画布和坐标系

首先导入matplotlib库，新建一个画布

import matplotlib.pyplot as plt #导入matplotlib库
#  1.建立画布
fig = plt.figure(figsize = (6,4))#将画布的宽度设置为6，高度设置为4

然后建立坐标系，可以使用add_subplot函数、plt.subplot函数或plt.subplots函数建立坐标系。

先是add_subplot函数

import matplotlib.pyplot as plt #导入matplotlib库
#  1.建立画布
fig = plt.figure(figsize = (6,4))#将画布的宽度设置为6，高度设置为4
#  2.add_subplot建立坐标系
ax1 = fig.add_subplot(1,1,1)#创建1个坐标系
plt.show()

 如果想同时绘制几个子图，可以

#  2.add_subplot建立坐标系
ax2 = fig.add_subplot(1,2,1)
ax3 = fig.add_subplot(1,2,2)
plt.show()

 会产生1*2个坐标系，1行2列的。

然后plt.subplot函数，将画布分成2*2个区域，然后在第1和2个区域里绘图

x = list(range(6))
y = [i**2 for i in x]
fig1 = plt.subplot(2,2,1)
fig2 = plt.subplot(2,2,2)
fig1.scatter(x,y)
fig2.plot(x,y)
plt.show()

 plt.subplots函数同理。

x = list(range(6))
y = [i**2 for i in x]
fig,ax = plt.subplots(2,2)
ax[0,0].scatter(x,y)
ax[0,1].scatter(y,x)
ax[1,0].plot(x,y)
ax[1,1].plot(y,x)
plt.show()

2.设置坐标轴标题

使用xlable()和ylabel()方法，使用fontsize 、color 、fontweight对字体大小、颜色、加粗进行调整

import matplotlib.pyplot as plt #导入matplotlib库
fig = plt.figure(figsize = (4,3))
x = list(range(12))
y = [i**2 for i in x]
fig = plt.plot(x,y)
plt.xlabel('x',fontsize = 20,color = 'r', fontweight = 'bold')
plt.ylabel('y')
plt.show()

 

3.设置坐标刻度值

使用plt.xticks()和plt.yticks()来设置

plt.xticks(x,[0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11])

 

 如果不需要显示刻度值，可以

plt.xticks([])
plt.yticks([])

4.设置坐标轴范围

使用xlim()和ylim()进行设置最小值最大值

plt.ylim(0,200)

5.设置网格线

默认是关闭的，可以通过设置b=True，将x轴和y轴的网格线全部打开

plt.grid(b=True)

可以通过设置axis的值来控制打开哪个坐标轴的网格线，还可以设置网格线的线型和线宽

 

plt.grid(b=True,axis = 'y',linestyle = 'dashed',linewidth = 1,color = 'r')

6.设置图例

通过label传入图例名，plt.legend()将图例显示出来

fig = plt.plot(x,y,label = 'y = 2*x+1')
plt.xlabel('x')
plt.ylabel('y')
plt.legend()
plt.show()

 

 只需要显示一个图例的话还可以

plt.legend(['y = 2*x+1'])

调整图例位置可以使用loc参数

plt.legend(loc='upper right')

best 表示自动分配最佳位置，还有upper left、lower left、lower right、right、center、center left、center right,lower center、 upper center。 

7.设置图表标题

plt.title('y=2*x+1',loc = 'center')

8.设置数据标签

plt.text(1,1,1)
plt.plot(1,1,'*')

 9.保存图表

plt.savefig('figure.png')