您可以使用为此找到:

[rowIdx,colIdx] = find(output==1);
output(output==1) = colIdx;