Python 绘制三角形信号

import numpy as np
import matplotlib.pyplot as plt

#幅度为h，宽度为w,斜度为h/0.5w的三角信号
def Triangle(x,w,h):   
    if x < 0.5*w:
        r= 2*h*x/w
    else:
        r= -2*h*x/w + 2*h/w*w
    return r


#设置宽度幅度，绘制时域区间，计算对应函数值，并作图
width1 = 2
height = 1
x1=np.linspace(0,width1,width1+1)
y1=np.array([Triangle(t,width1,height) for t in x1])
plt.plot(x1,y1)
plt.show()