Устранение части массивов в matlab

Question

Предположим, что у нас есть массив 100 * 4.

У нас также есть массив 100 * 1 из 1 и 0. Предположим, что есть n 1.

Мы хотим создать п * 4 массив из массива в 100 * 4, где мы включаем только столбцы, для которых второй массив является 1.

Один из способов сделать это через двойной цикл. Есть ли более простой метод?

Таким образом, мы имеем

A = [ [ 332 44 33 22 33 55 33 211 ..... 
     [ 823 44 12 98 19 23 32 911 ..... 
     .... 
     .... 
    ] 

и

B = [1 0 0 1 0 0 0 ....] 

и мы хотим

C = [ [ 332 22 ... 
     [ 823 98 ... 
     .... 
     .... 
    ] 

Answer 1

Вы должны использовать logical indexing:

C = A(:, B==1); 

Answer 2

Сначала вы repmat логический вектор так, что он имеет точно такой же размер, что и матрица A.

idx2keep = repmat(b, [1 4]); % Or [4 1] depending on if it's a col or row vector 

Тогда вы можете просто индексировать их с

B = A(idx2keep) 

вы можете превратить его в вектор-столбец:

B = B(:) 

Это должно сделать работу. В следующий раз, пожалуйста, всегда отправляйте код или нотацию, чтобы было легче и понятнее ответить на этот вопрос.