MATLAB中如何将指定for循环改写为无循环向量形式？

存在完全无循环的向量化实现，运行效率远高于原for循环，且计算结果与原代码完全一致。

原循环本质是构造一个1000行、3000列的矩阵A，规则如下：

• 输入维度：data为1000×1列向量，center1为1×3行向量
• 逐行操作规则：对第i行，仅在(i-1)*3+1至i*3的连续3列位置，填入计算值2*(data(i) - center1)，该行其余位置为0
• 最终矩阵的非零元素呈对角分块分布，其余位置全部为0

1. 稀疏矩阵高效构造（最优方案） #

由于A的绝大多数位置为0，使用MATLAB原生sparse函数直接构造，内存占用仅为稠密矩阵的1/1000左右，计算速度最快：

% 预计算所有非零元素值，利用MATLAB隐式扩展完成广播计算
B = 2 * (data - center1);
% 构造非零元素的行索引：每个块占3个相同行号
row_ind = repelem(1:size(data,1), 3);
% 构造非零元素的列索引：每个块占连续3列
col_ind = reshape( (3*(0:size(data,1)-1))' + (1:3), 1, [] );
% 按行优先顺序展平B，匹配行列索引的排布顺序
val = reshape(B', 1, []);
% 生成矩阵A
A = sparse(row_ind, col_ind, val, size(data,1), 3*size(data,1));

老版本兼容说明：如果使用R2016b之前的MATLAB版本，不支持隐式广播，可将B的计算替换为B = 2*bsxfun(@minus, data, center1);，结果完全一致。

2. 稠密全矩阵构造 #

如果需要生成非稀疏的全存储矩阵，可使用blkdiag函数直接构造对角分块矩阵：

B = 2 * (data - center1);
% 将B按行拆分为单个行块
blocks = mat2cell(B, ones(1, size(data,1)), 3);
% 直接拼接为对角分块矩阵
A = blkdiag(blocks{:});

3. 预分配稠密矩阵写入 #

如果需要手动预分配稠密矩阵，可直接通过线性索引批量写入非零值，无需循环：

n = size(data,1);
A = zeros(n, 3*n);
B = 2 * (data - center1);
% 生成所有非零位置的行列坐标
row_pos = (1:n)';
col_pos = 3*(row_pos-1) + [1,2,3];
% 按对应位置批量赋值
A(sub2ind(size(A), repmat(row_pos,1,3), col_pos)) = B;

• 三类方法的计算结果与原循环代码完全等价
• 稀疏构造方案适合大规模矩阵场景，内存占用极低
• 稠密构造方案适合后续需要频繁全矩阵运算的场景，代码简洁，运行效率接近底层C实现

内容的提问来源于stack exchange，提问作者SEU