你可以用简单的矩阵乘法来解决这个问题.
result = M * (1:size(M, 2)).';
3
1
2
1
3
这是通过将您的M x 3矩阵与3 x 1阵列相乘来实现的,其中3×1的元素仅为[1; 2; 3].简而言之,对于M的每一行,使用3×1阵列执行逐元素乘法.只有M行中的1将在结果中产生任何结果.然后将该元素乘法的结果相加.因为每行只有一个“1”,所以结果将是1所在的列索引.
所以例如M的第一行.
element_wise_multiplication = [0 0 1] .* [1 2 3]
[0, 0, 3]
sum(element_wise_multiplication)
3
更新
根据@reyryeng和@Luis提供的解决方案,我决定进行比较,看看各种方法的性能如何比较.
为了设置测试矩阵(M),我创建了一个原始问题中指定形式的矩阵,并改变了行数.使用randi([1 nCols],大小(M,1))随机选择哪一列具有1.使用timeit分析执行时间.
当使用M类型double(MATLAB的默认值)运行时,您将获得以下执行时间.
如果M是逻辑的,则矩阵乘法由于在矩阵乘法之前必须将其转换为数字类型的事实而受到命中,而其他两个具有一点性能改进.
这是我使用的测试代码.
sizes = round(linspace(100, 100000,