Matlab移动平均线

Question

我试图创建一个名为moving_average.m 的函数，它有一个数组作为参数 任意长度，并从三个连续值计算移动平均值。结果应该返回一个数组。这是我当前的代码：

function [output] = moving_average(inputarray)
    sz = size(inputarray);
    output = zeros(sz);
    for i=1:sz
        output(i,:) = (inputarray(i) + inputarray(i+1) + inputarray(i+2))/3;
    end
    
end

现在，它只设置输出数组的第一个元素，其他元素为零。有人知道如何解决这个问题吗？

Answer 1

size 函数的输出是一个 Nx1 数组，其中 N 是输入的维数。

inputarray = rand(1,20);
sz = size(inputarray)

输出：

sz =

    1     20

语句i = 1:sz 需要一个标量值。因此，Matlab 只查看sz 的第一个元素，对于 1xN 数组，它是 1。如果您使用 Nx1 行数组，它将运行 N 次，但随后您将收到错误 (Index exceeds array bounds.)，因为它将尝试访问不存在的行 N+1 和 N+2。这就是您需要使用 for i=1:sz-2 的原因，如 Martin 所示。

length 函数等价于max(size(A))。这很方便，因为它可以无缝处理行向量和列向量。

Answer 2

因为您正在执行 3 点移动平均，for 循环正在尝试访问值 i、i+1 和 i+2，但它们不存在。这就是我将如何重写函数moving_average：

function [output] = moving_average(inputarray)
    len_z = length(inputarray);
    output = zeros(len_z-2,1);
    for i=1:len_z-2
        output(i) = (inputarray(i) + inputarray(i+1) + inputarray(i+2))/3;
    end

end

然后用以下方法测试它：

X = rand(20,1);

moving_average(X)

我使用20x1 向量作为输入X，但它适用于任何nx1 或1xn。 moving_average(X) 的输出将是 (n-2)x1。