matplotlib画图

matplotlib是python最常用的绘图工具，过段时间不用就会忘记怎么用，所以还是写篇文章加深下印象。

1. 绘制曲线需要确定x和y：（最基础版）

import numpy as np
import matplotlib.pyplot as plt

x = np.arange(0.1,0.9,0.1)
y = np.array([3,7,2,6,2,6,1,8])

plt.plot(x,y,'r',lw=1) # 颜色为红，线宽为1个pixel

plt.show()

plt.savefig("fig1.png") # 保存为图片

 

2. 增加点边边角角：（横纵坐标标签、网格线、标题）

import numpy as np
import matplotlib.pyplot as plt

x = np.arange(0.1,0.9,0.1)
y = np.array([3,7,2,6,2,6,1,8])

plt.plot(x,y,'r',lw=1)# 颜色为红，线宽为1个pixel

plt.xlabel('ratio')
plt.ylabel('amount')
plt.grid() # 增加网格
plt.title('my_plot')

plt.show()

3. 调整x, y轴范围，调整刻度：

import numpy as np
import matplotlib.pyplot as plt

x = np.arange(0.1,0.9,0.1)
y = np.array([3,7,2,6,2,6,1,8])

plt.xlim(0.1, 0.8) # x 范围
plt.ylim(0, 10)
plt.xticks(np.arange(0.1,0.9,0.1)) # x轴刻度
plt.yticks(np.arange(0,10,1))

plt.plot(x,y,'r',lw=1)# 颜色为红，线宽为1个pixel

plt.xlabel('ratio')
plt.ylabel('amount')
plt.grid() # 增加网格
plt.title('my_plot')

plt.show()

 

4. 画多个折现，并设置铭牌legend：

import numpy as np
import matplotlib.pyplot as plt

x = np.arange(0.1,0.9,0.1)
y1 = np.array([3,7,2,6,2,6,1,8])
y2 = np.array([2,1,3,1,5,6,5,1])

plt.xlim(0.1, 0.8) # x 范围
plt.ylim(0, 10)
plt.xticks(np.arange(0.1,0.9,0.1)) # x轴刻度
plt.yticks(np.arange(0,10,1))


l1, = plt.plot(x,y1,'r',lw=1) # 给曲线编号为l1
# l1, = plt.plot(x, y1, 'r', lw=1, '--') # 画虚线
l2, = plt.plot(x,y2,'b',lw=1)

plt.xlabel('ratio')
plt.ylabel('amount')
plt.grid() # 增加网格
plt.title('my_plot')
plt.legend(handles = [l1, l2], labels = ['plot1', 'plot2'], loc = 'best')
#plt.savefig('my_plot.png')
plt.show()

加'--'可以画出虚线～

留个坑，日后完善，持续更新