correlation for multi-dimensional arrays

Question

Niko 2011-9-14

0
链接

此问题的直接链接

https://ww2.mathworks.cn/matlabcentral/answers/15884-correlation-for-multi-dimensional-arrays

回答： Mustapha Adamu 2018-12-10

Hi everyone,

I need to compute correlation coefficients - lots of them. I have two three-dimensional arrays (frequency x time x observations) and I want to compute correlations between the two arrays along the third dimension. The result I need is a two-dimensional array of correlation coefficients (frequency x time). If I understand the corr function correctly, corr is only for column vectors. If I loop over my other two dimensions, I can of course compute the correlation for each time-frequency point separately, but this is very slow.

Is there a way to compute correlation coefficients for multi-dimensional arrays along an arbitrary dimension, or any other way to speed up the computation of correlations?

Thanks!

1 个评论
显示 -1更早的评论隐藏 -1更早的评论

Raj 2014-8-27

Can you please provide formulas (math models) for the multi-dimensional correlation?

请先登录，再进行评论。

请先登录，再回答此问题。

Answer 1

David Young 2011-9-15

2
链接

此回答的直接链接

https://ww2.mathworks.cn/matlabcentral/answers/15884-correlation-for-multi-dimensional-arrays#answer_21553

在 MATLAB Online 中打开

If you don't have NaNs in the data, and you want the standard Pearson coefficient, then you could try applying the formula for correlation directly, like this:

% Synthetic data for testing
a = rand(10, 10, 100);
b = rand(10, 10, 100);
b(1, 1, :) = 3 * a(1, 1, :) - 2; % r(1,1) should be + 1;
b(1, 2, :) = -17 * a(1, 2, :) + 8; % r(1,2) should be - 1;
% rest of r should be random between +1 and -1
% Compute correlations on third dimension
% Remove means 
az = bsxfun(@minus, a, mean(a,3));
bz = bsxfun(@minus, b, mean(b,3));
% Standard Pearson correlation coefficient formula
a2 = az .^ 2;
b2 = bz .^ 2;
ab = az .* bz;
r = sum(ab, 3) ./ sqrt(sum(a2, 3) .* sum(b2, 3));