Python图例与坐标设置 - 绘制多条曲线

相关知识

 

1. 设置图例；
2. 设置坐标轴及标题；
3. 绘制多条曲线。

设置图例

在 matplotlib 中，可通过legend函数设置图例，调用方式如下：

1. legend()                       #以默认形式设置图例
2. legend(labels)               #标记已有的绘
3. legend(handles, labels)  #明确定义图例中的元素

 

设置坐标轴及标题

在使用 matplotlib 模块画坐标图时，往往需要对坐标轴设置很多参数，这些参数包括横纵坐标轴范围、坐标轴刻度大小、坐标轴名称等等，在 matplotlib 中包含了很多函数，用来对这些参数进行设置。

• plt.xlim、plt.ylim用于设置横纵坐标轴范围；
• plt.xlabel、plt.ylabel用于设置坐标轴名称；
• plt.xticks、plt.yticks用于设置坐标轴刻度；
• plt.title用于设置图像标题。

上面的plt表示 matplotlib.pyplot 模块，即在程序中以import matplotlib.pyplot as plt方式导入 matplotlib.pyplot 模块。

 

绘制多条曲线

绘制多条曲线有两种情况：
第一种是在同一坐标系上绘制多条曲线，能够清楚地看到多条曲线的对比情况。可通过直接叠加使用plot进行绘制。

代码示例：

#导入库文件
import matplotlib.pyplot as plt
import numpy as np

# ********** Begin *********#
#定义图像函数
def f1(t):
    y1 = t**2*np.exp(-t**2)
    return y1
def f2(t):
    y2 = t**4*np.exp(-t**2)
    return y2
#设置图像属性
t = np.linspace(0,3,num = 50)
y1 = f1(t)
y2 = f2(t)
#绘制图像
plt.plot(t,y1,'r--')
plt.plot(t,y2,'bo-')
#设置图表属性
plt.legend(['y1','y2'])#图例
plt.xlabel('x')#设置x,y轴标记
plt.ylabel('y')
plt.title('Plotting two curves in the same plot')#设置图像标题
#保存图像
plt.savefig('fig3.png')
# ********** End **********#

运行结果如下：