Matlab 可变大小数组

Question

我对 Matlab 很陌生，对数组的使用感到完全不知所措。以下 C++ 代码在 Matlab 中最有效的实现是什么？

A = std::vector<double>();

for (int i = 0; i < 100; i++) {
  if (complicatedBoolFunction(i)) {
    A.push_back(i);
  }
}

编辑：效率是指使用尽可能少的资源来增长数组 A - 即避免将其复制粘贴到临时内存中

Answer 1

此类 C++ 代码的最有效实现是

i = 0:99;
A = i(complicatedBoolFunction(i));

无论如何，您都可以通过连接来增长一个数组，这通常是（或曾经）不推荐的，如下所示

A = [];

for i = 0:99
  if (complicatedBoolFunction(i))
    A = [A i];
  end
end

或much more efficiently 像这样：

A = [];

for i = 0:99
  if (complicatedBoolFunction(i))
    A(end + 1) = i;
  end
end

Answer 2

你可以通过两种方式做到这一点

1. 预分配最大尺寸，并删除未使用的元素。这样做的好处是在经常满足条件的情况下预分配内存...

   A = NaN(100,1)
   for ii = 0:99
       if rand > 0.5    % some condition
           A(ii+1) = ii; % some value
       end
   end
   A(isnan(A)) = []; % remove unused elements
   

2. 附加到数组。如果不太可能追加，这可以避免使 A 太大...

   A = []; % empty array
   for ii = 0:99
       if rand > 0.5 % some condition
           A(end+1, 1) = ii; % some value. Equivalent to 'A = [A; ii];'
       end
   end
   

一种更好、更类似于 Matlab 的方法是将条件函数向量化。这样可以避免循环和分配问题...

ii = 0:99;
A = ii(rand(100, 1) > 0.5);

您可以使用任何您喜欢的布尔函数作为索引数组，只要它返回的逻辑数组的元素数量与您要索引的数组（此处为ii）或元素的整数索引相同即可选择。