快速，向量化方式将二进制矩阵转换为最后一个非零索引的向量

[poster]

假设在MATLAB中，我有一个矩阵A，其元素为0或1。

如何以更快的矢量化方式获取每列最后一个非零元素的索引的向量？

我可以做

[B, I] = max(cumsum(A));

并使用I ，但是有更快的方法吗？ （我假设即使将0和1加起来也将花费一些时间）。

编辑：我想我向量化的速度远远超过了我所需要的速度-Fooz先生的循环很棒，但即使快速，MATLAB中的每个循环似乎也花费了我很多调试时间。


回答：
如Fooz先生所示 ，使用新版本的MATLAB，for循环现在可以很快。但是，如果您真的想拥有紧凑的矢量化代码，建议您尝试以下操作：

[B,I] = max(flipud(A)); I = size(A,1)-I+1; 这比基于CUMSUM的答案要快，但仍然不如Fooz先生的循环选项快。

需要考虑的另外两件事：

• 对于完全没有列的列，您想要获得什么结果？通过以上给出的选项，我相信在这种情况下，您将获得size（A，1）的索引（即A中的行数）。作为您的选择，我相信您会在这种情况下获得1，而Fooz先生的“嵌套循环”选项将为您提供0。
  
• 这些不同选项的相对速度可能会根据A的大小和期望的非零数目而有所不同。
  

更多&回答...