作者:hro5028136 | 来源:互联网 | 2023-12-12 20:24
本文介绍了在使用python的matplotlib库画水平直线时可能遇到的问题,并提供了解决方法。通过导入numpy和matplotlib.pyplot模块,设置绘图对象的宽度和高度,以及使用plot函数绘制水平直线,可以解决该问题。
想要的图像如下:
一开始是这样画的:
import numpy as np #使用import导入模块numpy,并简写成np
import matplotlib.pyplot as plt #使用import导入模块matplotlib.pyplot,并简写成plt
plt.figure(figsize=(8,4)) #设置绘图对象的宽度和高度t = np.arange(0,1.1,0.1)
theta = 30+15*(t**2)
theta_v = 30*t
theta_accel = 30 plt.plot(t,theta,label="$theta$",color="red",linewidth=2)
plt.plot(t,theta_v,label="$thetaV$",color="green",linewidth=2)
#plt.plot(t,theta_accel,label="$thetaAccel$",color="blue",linewidth=2)t = np.arange(1,3.1,0.1)
theta = 45+30*(t-1)
theta_v = 30
theta_accel = 0 plt.plot(t,theta,label="$theta$",color="red",linewidth=2)
#plt.plot(t,theta_v,label="$thetaV$",color="green",linewidth=2)
#plt.plot(t,theta_accel,label="$thetaAccel$",color="blue",linewidth=2)t = np.arange(3,4.1,0.1)
theta = 120-15*((4-t)**2)
theta_v = 30*(4-t)
theta_accel = -30 plt.plot(t,theta,label="$theta$",color="red",linewidth=2)
plt.plot(t,theta_v,label="$thetaV$",color="green",linewidth=2)
#plt.plot(t,theta_accel,label="$thetaAccel$",color="blue")plt.ylim(-40,200) #使用plt.ylim设置y坐标轴范围
plt.xlim(-1,5)
plt.xlabel("Time(s)") #用plt.xlabel设置x坐标轴名称
plt.legend(loc='upper left') #设置图例位置
plt.grid(True)
plt.show()
#plt.plot(t,theta_accel,label="$thetaAccel$",color="blue",linewidth=2)
可以发现当theta_accel为常数时 plot失效,无法画出图像。
因为theta_accel = -30 不含变量t,改为:
theta_accel = -30 +t*0
则函数能够画出想要画的图像。
修改完,代码如下:
"""niku 习题5.5"""
import numpy as np #使用import导入模块numpy,并简写成np
import matplotlib.pyplot as plt #使用import导入模块matplotlib.pyplot,并简写成plt
plt.figure(figsize=(8,4)) #设置绘图对象的宽度和高度t = np.arange(0,1.1,0.1)
theta = 30+15*(t**2)
theta_v = 30*t
theta_accel = 30+t*0 plt.plot(t,theta,label="$theta$",color="red",linewidth=2)
plt.plot(t,theta_v,label="$thetaV$",color="green",linewidth=2)
plt.plot(t,theta_accel,label="$thetaAccel$",color="b")t = np.arange(1,3.1,0.1)
theta = 45+30*(t-1)
theta_v = 30+t*0
theta_accel = 0 +t*0 plt.plot(t,theta,label="$theta$",color="red",linewidth=2)
plt.plot(t,theta_v,label="$thetaV$",color="green",linewidth=2)
plt.plot(t,theta_accel,label="$thetaAccel$",color="b",linewidth=2)t = np.arange(3,4.1,0.1)
theta = 120-15*((4-t)**2)
theta_v = 30*(4-t)
theta_accel = -30 +t*0 plt.plot(t,theta,label="$theta$",color="red",linewidth=2)
plt.plot(t,theta_v,label="$thetaV$",color="green",linewidth=2)
plt.plot(t,theta_accel,label="$thetaAccel$",color="b")plt.ylim(-40,125) #使用plt.ylim设置y坐标轴范围
plt.xlim(-1,5)
plt.xlabel("Time(s)") #用plt.xlabel设置x坐标轴名称
'''设置图例位置'''
#plt.legend(loc='upper right') #设置图例位置
plt.grid(True)
plt.show()
生成图像如下:
成功!!!