在Matlab中返回索引的函数

Question

如何在Matlab中编写一个函数,该函数采用每列中具有单个1值的矩阵并返回此1的索引.

例如:如果输入为x = [0 0 1; 1 0 0; 0 1 0],则返回索引= [2 3 1]

Answer 1

find 确实是要走的路

[indices,~] = find(x);

如果你想更加隐秘地做,或者find出于某种原因讨厌,你也可以使用cumsum:

indices = 4 - sum(cumsum(x,1),1);