python 实现正弦波、三角波、方波、锯齿波

最新推荐文章于 2024-11-15 05:30:00 发布

没有水杯和雨伞的工科男

最新推荐文章于 2024-11-15 05:30:00 发布

阅读量1.3w

点赞数 18

CC 4.0 BY-SA版权

分类专栏： Python

本文链接：https://blog.youkuaiyun.com/weixin_42066185/article/details/103600532

Python 专栏收录该内容

53 篇文章

订阅专栏

本文介绍了一种使用Python的Numpy库生成不同波形的方法，包括正弦波、三角波、方波和锯齿波。通过定义函数并设置周期、精度等参数，可以灵活地创建各种波形，为工程测试提供了便利。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

直接上代码，本人由于工程测试需要，使用numpy的库写了下面的代码产生函数，

大家点赞，转发哦转载请注明出处

import numpy as np
import matplotlib.pyplot as plt

def sin_wave(start,end,zhouqi,midu):
    '''

    :param start: the fist value of the wave
    :param end:  the end value of the wave
    :param zhouqi:  the zhouqi range of the wave
    :param midu:  every zhouqi, there are how many points in this zhouqi
    :return: the x array and the y array
    '''

    # 根据numpy的sin函数，生成对应的y的坐标。
    # y = np.sin(x)

    xout=[]
    yout=[]
    for i in range(start,end,zhouqi):
        x = np.arange(i, i + zhouqi,  midu)
        # y = np.where(x<start+0.5, x-start, 0)
        y = np.sin(x)

        xout = np.append(xout, x)
        yout = np.append(yout, y)
    return xout,yout





def triangle_wave(start,end,zhouqi,midu):
    '''

    :param start: the fist value of the wave
    :param end:  the end value of the wave
    :param zhouqi:  the zhouqi range of the wave
    :param midu:  every zhouqi, there are how many points in this zhouqi
    :return: the x array and the y array
    '''

    xout=[]
    yout=[]
    for i in range(start,end,zhouqi):
        x = np.arange(i, i + zhouqi,  midu)
        # y = np.where(x<start+0.5, x-start, 0)
        y = np.where(x >= i + zhouqi/2, i + zhouqi - x, x - i)

        xout = np.append(xout, x)
        yout = np.append(yout, y)
    return xout,yout



def square_wave(start,end,zhouqi,midu):
    '''

    :param start: the fist value of the wave
    :param end:  the end value of the wave
    :param zhouqi:  the zhouqi range of the wave
    :param midu:  every zhouqi, there are how many points in this zhouqi
    :return: the x array and the y array
    '''
    xout = []
    yout = []
    for i in range(start,end,zhouqi):
        x = np.arange(i, i + zhouqi,  midu)
        # y = np.where(x<start+0.5, x-start, 0)
        y = np.where(x >= i+ zhouqi/2, 1,0)

        xout = np.append(xout, x)
        yout = np.append(yout, y)
    return xout, yout



def swatooth_wave(start,end,zhouqi,midu):
    '''

    :param start: the fist value of the wave
    :param end:  the end value of the wave
    :param zhouqi:  the zhouqi range of the wave
    :param midu:  every zhouqi, there are how many points in this zhouqi
    :return: the x array and the y array
    '''
    xout = []
    yout = []
    for i in range(start,end,zhouqi):
        x = np.arange(i, i + zhouqi, midu)
        # y = np.where(x<start+0.5, x-start, 0)
        y = np.where(x >= i, i + zhouqi - x, x - i)

        xout = np.append(xout, x)
        yout = np.append(yout, y)
    return xout, yout


if __name__=='__main__':



    x,y=sin_wave(1,10,2,0.0001)
    # for  i in range()
    plt.plot(x,y)
    plt.show()
    x,y=triangle_wave(1,10,2,0.0001)
    # for  i in range()
    plt.plot(x,y)
    plt.show()
    x,y=square_wave(1,10,2,0.0001)
    # for  i in range()
    plt.plot(x,y)
    plt.show()
    x,y=swatooth_wave(1,10,2,0.0001)
    # for  i in range()
    plt.plot(x,y)
    plt.show()

将上面的代码简化一层

import numpy as np
import matplotlib.pyplot as plt

def sin_wave(start,zhouqi,midu,xdecimals,ydecimals):
    '''

    :param start: the fist value of the wave
    :param end:  the end value of the wave
    :param zhouqi:  the zhouqi range of the wave
    :param midu:  every zhouqi, there are how many points in this zhouqi
    :param xdecimals:  x decimals
    :param ydecimals: y decimals
    :return: the x array and the y array
    '''

    # 根据numpy的sin函数，生成对应的y的坐标。
    # y = np.sin(x)

    xout=[]
    yout=[]
    # for i in range(start,end,zhouqi):
    x = np.around(np.arange(start,start+zhouqi,midu),decimals=xdecimals)
    # y = np.where(x<start+0.5, x-start, 0)
    y = np.around(np.sin(x),decimals=ydecimals)

        # xout = np.append(xout, x)
        # yout = np.append(yout, y)
    return x,y





def triangle_wave(start,zhouqi,midu,xdecimals,ydecimals):
    '''

    :param start: the fist value of the wave
    :param end:  the end value of the wave
    :param zhouqi:  the zhouqi range of the wave
    :param midu:  every zhouqi, there are how many points in this zhouqi
    :return: the x array and the y array
    '''

    xout=[]
    yout=[]
    x = np.around(np.arange(start, start + zhouqi,  midu),decimals=xdecimals)
    # y = np.where(x<start+0.5, x-start, 0)
    y =np.around(np.where(x >= start + zhouqi/2, start + zhouqi - x, x - start),decimals=ydecimals)

    return x,y



def square_wave(start,zhouqi,midu,xdecimals,ydecimals):
    '''

    :param start: the fist value of the wave
    :param end:  the end value of the wave
    :param zhouqi:  the zhouqi range of the wave
    :param midu:  every zhouqi, there are how many points in this zhouqi
    :return: the x array and the y array
    '''
    xout = []
    yout = []
    x =np.around(np.arange(start, start + zhouqi,  midu),decimals=xdecimals)
    # y = np.where(x<start+0.5, x-start, 0)
    y =np.around(np.where(x >= start+ zhouqi/2, 1,0),decimals=ydecimals)

    return x, y



def swatooth_wave(start,zhouqi,midu,xdecimals,ydecimals):
    '''

    :param start: the fist value of the wave
    :param end:  the end value of the wave
    :param zhouqi:  the zhouqi range of the wave
    :param midu:  every zhouqi, there are how many points in this zhouqi
    :return: the x array and the y array
    '''
    xout = []
    yout = []
    x =np.around(np.arange(start, start + zhouqi, midu),decimals=xdecimals)
    # y = np.where(x<start+0.5, x-start, 0)
    y =np.around(np.where(x >= start, start + zhouqi - x, x - start),decimals=ydecimals)

    return x,y


if __name__=='__main__':



    x,y=sin_wave(1,10,2,0.1,xdecimals=2,ydecimals=5)
    # for  i in range()
    plt.plot(x,y)
    plt.show()
    # x,y=triangle_wave(1,10,2,0.0001)
    # # for  i in range()
    # plt.plot(x,y)
    # plt.show()
    # x,y=square_wave(1,10,2,0.0001)
    # # for  i in range()
    # plt.plot(x,y)
    # plt.show()
    # x,y=swatooth_wave(1,10,2,0.0001)
    # # for  i in range()
    # plt.plot(x,y)
    # plt.show()
    #