MATLAB：无需循环将矩阵各列指定前N行置零的方法

嘿，我来给你分享几个不用显式循环的高效方案，完美替代你现在的for循环实现，而且在处理大矩阵时性能会更出色！

方法一：利用隐式扩展构造逻辑掩码（MATLAB R2016b+） #

这是最简洁高效的写法，借助MATLAB的隐式扩展功能，不用手动重复矩阵就能完成元素级比较：

mat = ones(100,5);
startRows = {10,10,30,10,40};

% 第一步：把元胞数组转换成普通数值向量
startRowsVec = cell2mat(startRows);
% 生成行索引列向量（行数和mat一致）
rowIdx = (1:size(mat,1))';
% 构造逻辑掩码：每个位置判断行号是否小于对应列的startRows值
mask = rowIdx < startRowsVec;
% 直接对掩码覆盖的位置置零
mat(mask) = 0;

原理很简单：rowIdx是100×1的列向量，startRowsVec是1×5的行向量，MATLAB会自动把它们扩展成100×5的矩阵进行逐元素比较，得到的mask矩阵里，所有需要置零的位置都是true，直接赋值即可。

方法二：兼容旧版本的bsxfun写法 #

如果你用的是MATLAB R2016b之前的版本，不支持隐式扩展，可以用bsxfun来实现同样的元素级操作：

mat = ones(100,5);
startRows = {10,10,30,10,40};

startRowsVec = cell2mat(startRows);
% 用bsxfun执行小于比较，生成掩码
mask = bsxfun(@lt, (1:size(mat,1))', startRowsVec);
mat(mask) = 0;

bsxfun的作用就是对两个维度不匹配的数组执行元素级运算，和隐式扩展的效果完全一致，只是写法稍显繁琐。

方法三：生成线性索引一次性赋值 #

如果你想直接定位所有需要置零的位置，也可以用arrayfun生成线性索引，然后批量置零：

mat = ones(100,5);
startRows = {10,10,30,10,40};

startRowsVec = cell2mat(startRows);
% 为每列生成需要置零的行对应的线性索引
idxList = arrayfun(@(col) sub2ind(size(mat), 1:(startRowsVec(col)-1), col), ...
                   1:size(mat,2), 'UniformOutput', false);
% 把所有索引合并后批量置零
mat([idxList{:}]) = 0;

不过这种方法底层其实还是循环（arrayfun本质是封装了循环），性能不如前两种向量化方法，更适合理解索引逻辑时使用。

你可以用isequal函数验证这些方法和原来的循环实现的结果是否一致，比如：

% 原循环实现的结果
matLoop = ones(100,5);
for c = 1 : 5
    matLoop(1:startRows{c}-1, c) = 0;
end
% 验证是否相等
disp(isequal(mat, matLoop)); % 应该输出1（true）

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