Matlab函数输出改为1D数组求助：新手编程遇单值输出问题

核心思路 #

你的函数当前只返回单个数值，本质是没有将多次计算的结果收集到数组中。只需要在函数里初始化数组、累积结果，最后返回这个数组即可。

具体实现步骤 #

• 初始化数组：如果知道结果的长度，优先预分配数组（效率更高），比如 result = zeros(1, 10);（表示创建1行10列的空数组）；如果不确定长度，先创建空数组 result = [];。
• 累积结果：
  • 预分配数组的情况：在循环或计算中，把每个结果放到对应位置，比如 result(i) = 计算得到的单个值;，i是当前结果的索引。
  • 动态扩容的情况：直接把新结果拼接到数组末尾，result = [result, 计算得到的单个值];。
• 返回数组：函数末尾将累积好的数组赋值给输出变量即可。

示例代码 #

比如一个计算1到n的立方值的函数，改造后返回1D数组：

function cube_array = calculateCubes(n)
    % 预分配1行n列的数组
    cube_array = zeros(1, n);
    for i = 1:n
        cube_array(i) = i^3;
    end
end

调用该函数：cubes = calculateCubes(4);，会得到结果 [1, 8, 27, 64]。

如果用向量化写法（更简洁高效，Matlab推荐），可以直接写成：

function cube_array = calculateCubes(n)
    cube_array = (1:n).^3;
end

注意事项 #

• 处理大量数据时，预分配数组比动态拼接的性能好很多，避免频繁扩容数组带来的开销。
• 尽量用Matlab的向量化运算替代循环，代码更简洁且运行更快。

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