如何使用多个循环在矩阵中进行计算？

Question

我有一个9 x 682矩阵（A）。 每36列对应一个新变量（总共17个变量）。 对于每个变量（36列），我想计算每6列的平均值（即，对于36列的每个块，第1、7、13、19、31列的平均值，依此类推。

我坚持使用几个For循环。

A_averaged = zeros(9,102);

for  i = 1:36:length(A)
    for j = i:i+35
        for k = j:6:j+1
            A_averaged(:,k) = mean(A(:,k), 2);
        end
    end
end

任何帮助深表感谢。

Answer 1

为了对此进行向量化，我将以一种智能的方式重塑数据，选择所需的列，然后计算均值。

% Reshape data to 3D with dimensions: 9 x 36 x (nGroups)
B = reshape(A, size(A, 1), 36, []);

% Now grab the columns of each chunk you want
C = B(:, 1:6:end, :);

% Now compute the mean along the first dimension and squeeze the result
result = squeeze(mean(C, 1));

您可以使用以下代码一行完成此操作：

result = squeeze(mean(reshape(A(:, 1:6:end), size(A, 1), 6, []), 1))

如果您想获取所有行和列的均值，则可以将数据重塑为45 x nGroups并沿第一维计算均值

result = mean(reshape(C, [], size(C, 3)), 1);

附带说明一下，您的9 x 682初始尺寸实际上不能被36列整除。 您是说9 x 612吗？