Matlab中基于向量元素无循环生成目标向量的实现方法
MATLAB下的无循环实现方案
当然有啦!不用手动写for循环的话,MATLAB里有好几种简洁高效的方式能实现你要的需求,下面给你两种常用的方案:
方案1:累加和+索引定位(高效首选,适合大规模数据)
这种方法完全靠向量操作实现,没有任何隐式或显式循环,性能拉满,特别适合处理大尺寸的A:
A = [1;2;3;4]; total_length = sum(A); % 计算每个分段的起始索引(比如第一个分段从1开始,第二个从1+1=2,第三个从2+2=4,以此类推) segment_starts = cumsum([1; A(1:end-1)]); % 初始化结果向量,长度是所有分段的总长度 B = zeros(total_length, 1); % 在每个分段的起始位置标记1 B(segment_starts) = 1; % 累加后就自动生成了每个分段的1到k序列 B = cumsum(B);
运行后B就是你要的[1;1;2;1;2;3;1;2;3;4]。原理很简单:每个分段的开头设为1,后续位置是0,累加的时候每到新分段就会从1重新开始计数,完美匹配你的需求。
方案2:arrayfun+vertcat(直观简洁,适合小数据场景)
如果追求代码可读性,不想搞复杂的索引计算,这种方法更直观——虽然arrayfun内部有隐式循环,但不用你手动写for,代码也很短:
A = [1;2;3;4]; % 遍历A的每个元素k,生成1到k的列向量,再把所有向量垂直拼接起来 B = vertcat(arrayfun(@(k) (1:k)', A, 'UniformOutput', false)...);
这里arrayfun会逐个处理A里的元素,每个元素k对应生成(1:k)'(转成列向量),UniformOutput=false让它返回一个细胞数组,最后用vertcat把细胞里的所有向量拼成长列向量。
小提示
- 如果你的
A是行向量,只需要先转成列向量(A = A')再用上面的方法就行。 - 方案1的时间复杂度是O(sum(A)),是理论最优的,处理大向量时比方案2快不少;方案2胜在代码短、容易理解,小数据场景下用起来特别顺手。
内容的提问来源于stack exchange,提问作者Meepo
相关产品推荐
相关产品推荐

