如何在矩阵的最后一列中移零

时间:2013-08-09 16:19:52

标签: matlab

我有一个像下面这样的矩阵 -

A=[1 1 1 1 1;
   0 1 1 1 2;
   0 0 1 1 3]

但我想将所有0放在行的末尾,所以A应该像 -

A=[1 1 1 1 1;
   1 1 1 2 0;
   1 1 3 0 0] 

我该怎么做? Matlab专家请帮助我。

7 个答案:

答案 0 :(得分:4)

你去吧。整个矩阵,没有循环,即使对于非连续的零也是如此:

A = [1 1 1 1 1; 0 1 1 1 2; 0 0 1 1 3];

At = A.'; %// It's easier to work with the transpose
[~, rows] = sort(At~=0,'descend'); %// This is the important part.
%// It sends the zeros to the end of each column
cols = repmat(1:size(At,2),size(At,1),1);
ind = sub2ind(size(At),rows(:),cols(:));
sol = repmat(NaN,size(At,1),size(At,2));
sol(:) = At(ind);
sol = sol.'; %'// undo transpose

像往常一样,对于在函数返回时不支持~符号的Matlab版本,请通过虚拟变量更改~,例如:

[nada, rows] = sort(At~=0,'descend'); %// This is the important part.

答案 1 :(得分:4)

更通用的例子:

A = [1 3 0 1 1;
     0 1 1 1 2;
     0 0 1 1 3]

% Sort columns directly
[~,srtcol] = sort(A == 0,2);
% Sorted positions 
sz  = size(A);
pos = bsxfun(@plus, (srtcol-1)*sz(1), (1:sz(1))'); % or use sub2ind

结果

B = A(pos)
B =
     1     3     1     1     0
     1     1     1     2     0
     1     1     3     0     0

答案 2 :(得分:2)

有很多方法可以做到这一点。一个快速的方式很容易就是这样:

a = [1 2 3 4 0 5 7 0];

idx=(find(a==0));

idx =

     5     8

b=a;   % save a new copy of the vector
b(idx)=[];    % remove zero elements

b =

     1     2     3     4     5     7

c=[b zeros(size(idx))]

c =

     1     2     3     4     5     7     0     0

您也可以修改此代码。

答案 3 :(得分:1)

如果您的零始终在一起,则可以使用circshift命令。这会将数组中的值移动指定的位数,并将从边缘运行的值包装到另一侧。您似乎需要为A中的每一行单独执行此操作,因此在上面的示例中,您可以尝试:

A(2,:) = circshift(A(2,:), [1 -1]);    % shift the second row one to the left with wrapping
A(3,:) = circshift(A(3,:), [1 -2]);    % shift the third row two to the left with wrapping

一般情况下,如果您的零始终位于A行的前面,您可以尝试以下内容:

for ii = 1:size(A,1)                              % iterate over rows in A
    numShift = numel(find(A(ii,:) == 0));         % assuming zeros at the front of the row, this is how many times we have to shift the row.
    A(ii,:) = circshift(A(ii,:), [1 -numShift]);  % shift it
end

答案 4 :(得分:1)

试试这个(快速破解):

for row_k = 1:size(A, 1)
   [A_sorted, A_sortmap] = sort(A(row_k, :) == 0, 'ascend');

   % update row in A:    
   A(row_k, :) = A(row_k, A_sortmap);
end

现在针对不支持~的MATLAB版本优化为垃圾lhs-identifier。

答案 5 :(得分:1)

@ LuisMendo的答案是鼓舞人心的优雅,但我无法让它工作(也许是一个matlab版本的东西)。以下(根据他的回答)为我工作:

Aaux = fliplr(reshape([1:numel(A)],size(A)));
Aaux(find(A==0))=0;
[Asort iso]=sort(Aaux.',1,'descend');
iso = iso + repmat([0:size(A,1)-1]*size(A,2),size(A,2),1);
A=A.';
A(iso).'

答案 6 :(得分:0)

我也问了这个问题并得到了一个超级优雅的答案(以上答案不一样): Optimize deleting matrix leading zeros in MATLAB