离散点曲线平滑

很多时候我们希望将画出的线段尽量变得平滑,这里说个简单的求均值平滑滤波方法。很简单,将一个数和其周围几个点相加然后求均值,这个均值做为平滑后的点。python代码:

 

def mean_filter(arr, step):
    """
    平滑滤波函数,输入是一个列表,输出是这个列表平滑之后的值。即取step个数的平均值
    :param arr:列表
    :param step:以多大步长取平均
    :return:平滑后的列表
    """
    new_arr = arr[:int(step / 2)]
    for kk in range(int(step / 2), len(arr) - step + int(step / 2)):
        new_arr.append(int(sum(arr[kk - int(step / 2):kk + step - int(step / 2)]) / step))
    new_arr.extend(arr[-(step - int(step / 2)):])
    return new_arr

这里是将step个点加一起作为平滑后输出。

平滑前:

平滑后:

(ps:上图需要对x,y轴都进行平滑。  图片来自[UESTC](http://www.uestc.edu.cn/),禁止转载使用)

你可能感兴趣的:(python)