查找非零行索引

Question

我有一個矩陣z（比如200x5），其中每行只有一個元素是非零的。如果不使用for循環，找到該索引的最有效方法是什麼？

例如：

z=[1 0 0;0 0 1]; 
a=findRow(z) 

其中應該顯示：

[1 3] 

Answer 1

原來， '查找' 是解決方案，不同之處在於我不得不轉Z：

[rowidx,~]=find(z'); 

編輯：必須首先轉置它，因爲matlab首先找到列。因此，如果不轉置，則行索引的順序關閉。

Answer 2

如果你有大多0元素的矩陣，你可以考慮使用sparse矩陣被設計爲空間高效在這種情況下：

z = sparse([1,0,0;0,0,1]); 

您仍然可以只使用find來得到你的答案

[~,idx] = find(z);