Сохранение ненулевых целых чисел из одной матрицы в другую

Question

Я пытаюсь создать цикл, который читает через матрицу (A), и сохраняет ненулевые значения в новую матрицу (w). Я не уверен, что не так с моим кодом.

function [d,w] = matrix_check(A) 
[nrow ncol] = size(A); 
total = 0; 
for i = 1:nrow 
    for j = 1:ncol 
     if A(i,j) ~= 0 
      total = total + 1; 
     end 
    end 
end 
d = total; 
w = []; 
for i = 1:nrow 
    for j = 1:ncol 
     if A(i,j) ~= 0 
      w = [A(i,j);w]; 
     end 
    end 
end 

Второй контур не работает (по крайней мере, он не распечатывает результаты w).

Answer 1

Второй цикл работает. Я предполагаю, что вы делаете:

>> matrix_check(A) 

И не:

>> [d, w] = matrix_check(A) 

MATLAB будет возвращать только первый выход, если не указано иное.

Как в стороне, вы можете выполнить свою задачу, используя MATLAB's logical indexing и воспользоваться преимуществами (намного быстрее, обычно) операций массива, а не циклов.

d = sum(sum(A ~= 0)); 
w = A(A ~= 0); 

Answer 2

Вы можете использовать nonzeros и nnz:

w = flipud(nonzeros(A)); %// flipud to achieve the same order as in your code 
d = nnz(A);