Вы можете использовать find
и возвращать только индексы столбцов, например
Y = [1 0 0; 0 1 0; 0 1 0];
[~, new_y] = find(Y); % output: [1; 2; 2] is the col indices of your 1s
Аналогично, вы можете вернуть индексы строк, если ваш ввод был транспонированным
[new_y, ~] = find(Y); % output: [1; 2; 3] is the row indices of your 1s