0

问题

我有一个矩阵 M 如下：

   M =  [1, 1, 0, 1, 0, 0, 0;
         0, 1, 1, 0, 1, 0, 0;
         0, 0, 1, 1, 0, 1, 0;
         0, 0, 0, 1, 1, 0, 1;
         1, 0, 0, 0, 1, 1, 0;
         0, 1, 0, 0, 0, 1, 1;
         1, 0, 1, 0, 0, 0, 1];

所有列中的总数为 21：

 Total_ones_in_cols = 21;

然后我预分配内存以查找 M 中每一列的行索引：

row_indices = zeros(1,Total_ones_in_cols);

下一步是查找所有列的行索引：

for i = 1:7
     Temp = find(M(:,i)); 
     row_indices = [row_indices, Temp.'];
end

问题

尽管对 row_indices 进行了预分配，MATLAB 仍然建议在循环内预分配 row_indices 以提高速度。有人可以解释为什么会这样吗？我的猜测是，由于我在循环中不断更改 row_indices 的大小，因此我预分配的先前内存被覆盖并被丢弃，这本质上意味着我所做的预分配变得无用。

4

1 回答 1

3

您已经根据最终大小正确分配了内存，row_indices但是您没有将循环中的结果存储在预分配的索引处，而是在最后附加它们。附加总是会终止预分配，因此 MATLAB 会告诉您进行预分配，因为row_indices循环末尾的大小是[1 42]（MATLAB 假设您想要这个结果）而不是[1 21]您预先分配（并实际寻找）的大小。

要修复您的代码，它将是：

row_indices = zeros(1,Total_ones_in_cols); 
for ii = 1:7 %i is for imaginary numbers; don't use i (and j) as variable name(s)
     Temp = find(M(:,ii)); 
     row_indices(3*ii-2 : 3*ii) = Temp;  %Storing results at their specific indices
end 
%This assumes that you already know that there are 3 ones in each column

我在您的问题中看到了标签。请注意，这find直接适用于矩阵，因此您可以通过以下方式避免循环（您应该为这样的简单任务执行此操作）：

[row_indices, ~] = find(M);
于 2018-06-29T11:01:43.263 回答