你可以预先分配一个随机大小的数组吗？

Question

相关代码的基本部分可以提炼成：

list=rand(1,x); % where x is some arbitrarily large integer
hitlist=[];
for n=1:1:x
    if rand(1) < list(n)
        hitlist=[hitlist n];
    end
end
list(hitlist)=[];

这个程序运行得很慢，我怀疑这就是原因，但我不知道如何修复它。 hitlist 的长度必然会以随机方式变化，因此我不能简单地预先分配适当大小的“零”。我打算将hitlist 和zeros 设置为列表的长度，但随后我将不得不删除所有多余的零，而且我不知道如何做到这一点而不会遇到同样的问题。

如何预分配随机大小的数组？

Answer 1

我不确定预分配“随机大小”，但您可以预分配大块，例如1e3，或者对您的用例有用：

list=rand(1,x); % where x is some arbitrarily large integer
a = 1e3; % Increment of preallocation
hitlist=zeros(1,a);
k=1; % counter
for n=1:1:x
    if rand(1) < list(n)
        hitlist(k) = n;
        k=k+1;
    end
    if mod(k-1,a)==0 % if a has been reached
        hitlist = [hitlist zeros(1,a)]; % extend
    end
end
hitlist = hitlist(1:k-1); % trim excess
% hitlist(k:end) = []; % alternative trim, but might error
list(hitlist)=[];

这不会是最快的，但至少比每次迭代递增快很多。确保选择a合适；您甚至可以使用memory 以某种方式基于可用的 RAM 量，然后修剪多余的部分，这样您就根本不需要执行循环内的技巧。

---

顺便说一句：MATLAB 以列为主，因此以这种方式运行矩阵更快。 IE。首先是第一列，然后是第二列，依此类推。对于一维数组，这无关紧要，但对于矩阵，它确实如此。因此我更喜欢使用list = rand(x,1)，即作为列。

对于这种特定情况，无论如何都不要使用这种循环方法，而是使用logical indexing：

list = rand(x,1);
list = list(list<rand(size(list)));