欢迎访问宙启技术站
智能推送

使用Python实现assign_moving_average()函数来计算数据的移动平均数

发布时间:2024-01-10 07:36:02

实现一个"assign_moving_average"函数,用Python编写。该函数接受两个参数:一个数据列表和一个窗口大小。它将计算移动平均数并将其作为新的列表返回。

移动平均数是指在给定窗口大小内计算数据平均值的方法。例如,如果数据是[1, 2, 3, 4, 5],窗口大小为3,则移动平均数将是[2, 3, 4]。 个元素2是[1, 2, 3]的平均值,第二个元素3是[2, 3, 4]的平均值,以此类推。

下面是"assign_moving_average"函数的实现代码:

def assign_moving_average(data, window_size):
    moving_averages = []
    for i in range(len(data) - window_size + 1):
        window = data[i:i+window_size]
        average = sum(window) / window_size
        moving_averages.append(average)
    return moving_averages

函数首先创建一个空列表"moving_averages"来存储移动平均数。然后,它使用一个循环来迭代数据列表,并对于每个窗口大小,计算窗口中数据的平均值,并将其添加到"moving_averages"列表中。

接下来,我们来实际使用该函数并看看它的输出。

data = [1, 2, 3, 4, 5, 6, 7, 8, 9, 10]
window_size = 3

moving_averages = assign_moving_average(data, window_size)
print(moving_averages)

在上述示例中,我们使用了一个长度为10的数据列表和窗口大小为3。函数的输出将是一个包含移动平均数的列表[2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0]。

上述代码在给定数据列表和窗口大小的情况下计算移动平均数,并打印结果。您可以根据自己的数据和窗口大小使用该函数来计算移动平均数。

总结一下,以上就是使用Python实现"assign_moving_average"函数来计算数据的移动平均数的方法。希望这对您有所帮助!