如何对数组

时间:2015-11-21 14:56:16

标签: matlab weighted-average

我有一个900×1的值向量(在MATLAB中)。每个9个连续的值应该被平均 - 没有重叠 - 结果是100×1的值向量。问题是应该根据[1 2 1;2 4 2;1 2 1]的加权向量对平均值进行加权。有没有有效的方法来进行平均化?我在MATLAB听说过conv函数;有用吗?

1 个答案:

答案 0 :(得分:0)

conv通过在您的数据中滑动内核来工作。但在你的情况下,你需要掩码跳过你的数据,所以我不认为conv会对你有用。

如果你想使用现有的MATLAB函数,你可以这样做(我必须假设你的加权矩阵只有一个维度):

kernel = [1;2;1;2;4;2;1;2;1];
in_matrix = reshape(in_matrix, 9, 100);
base = sum(kernel);
out_matrix = bsxfun(@times, in_matrix, kernel);
result = sum(out_matrix,1)/base;

我不知道是否有任何聪明的方法可以加快速度。 bsxfun允许单例扩展,但可能不是降维。

更快捷的方法是使用mex。在编辑器中打开一个新文件,粘贴以下代码并将文件另存为weighted_average.c

#include "mex.h"

void mexFunction(int nlhs, mxArray *plhs[], int nrhs, const mxArray *prhs[])
{
    double *in_matrix, *kernel, *out_matrix, base;              
    int niter;
    size_t nrows_data, nrows_kernel;                   

    /* Get number of element along first dimension of input matrix. */
    nrows_kernel = mxGetM(prhs[1]);    
    nrows_data = mxGetM(prhs[0]);

    /* Create output matrix*/
    plhs[0] = mxCreateDoubleMatrix((mwSize)nrows_data/nrows_kernel,1,mxREAL);

    /* Get a pointer to the real data */
    in_matrix = mxGetPr(prhs[0]);
    kernel = mxGetPr(prhs[1]);
    out_matrix = mxGetPr(plhs[0]);

    /* Sum the elements in weighting array */
    base = 0;
    for (int i = 0; i < nrows_kernel; i +=1)
    {
        base += kernel[i];
    }

    /* Perform calculation */
    niter = nrows_data/nrows_kernel;
    for (int i = 0; i < niter ; i += 1)
    {   
        for (int j = 0; j < nrows_kernel; j += 1)
        {
            out_matrix[i] += in_matrix[i*nrows_kernel+j]*kernel[j];
        }
        out_matrix[i] /= base;
    }    
}

然后在命令窗口中输入

mex weighted_average.c

使用它:

result = weighted_average(input, kernel);

请注意,inputkernel都必须是M x 1矩阵。在我的电脑上,第一种方法耗时0.0012秒。第二种方法耗时0.00007秒。这比第一种方法快一个数量级。