如何在Matlab中保存Matrix中的索引和值?

时间:2016-06-25 17:57:27

标签: matlab matrix

我有一个3x3矩阵,想要将索引和值保存到新的9x3矩阵中。例如A = [1 2 3 ; 4 5 6 ; 7 8 9],以便我得到矩阵x = [1 1 1; 1 2 2; 1 3 3; 2 1 4; 2 2 5; ...]使用我的代码我只能存储最后的值x = [3 3 9]

A = [1 2 3 ; 4 5 6 ; 7 8 9]; 
x=[];
 for i = 1:size(A)
   for j = 1:size(A)
      x =[i j A(i,j)]
   end
 end

感谢您的帮助

2 个答案:

答案 0 :(得分:3)

矢量化方法

这是避免循环的一种方法:

A = [1 2 3 ; 4 5 6 ; 7 8 9];
[ii, jj] = ndgrid(1:size(A,1), 1:size(A,2)); % row and column indices
vv = A.'; % values. Transpose because column changes first in the result, then row
x = [jj(:) ii(:) vv(:)]; % result

使用您的代码

您只能错过先前x

的连接
A = [1 2 3 ; 4 5 6 ; 7 8 9];
x = [];
for i = 1:size(A)
  for j = 1:size(A)
    x = [x; i j A(i,j)]; % concatenate new row to previous x
  end
end

另外两项建议:

  • 请勿使用ij作为variable names,因为这会影响虚构单位。
  • Preallocate x而不是在每次迭代中增长,以提高速度。

修改后的代码是:

A = [1 2 3 ; 4 5 6 ; 7 8 9];
x = NaN(numel(A),3); % preallocate
n = 0;
for ii = 1:size(A)
  for jj = 1:size(A)
    n = n + 1; % update row counter
    x(n,:) = [ii jj A(ii,jj)]; % fill row n
  end
end

答案 1 :(得分:0)

我开发了一种工作速度更快的解决方案。这是代码:

% Generate subscripts from linear index
[i, j] = ind2sub(size(A),1:numel(A));

% Just concatenate subscripts and values
x = [i' j' A(:)];

尝试一下,让我知道;)