如何返回包含最大标签的向量

Question

我有一个4列数组，我想获得一个向量，其中包含每行包含该行最大值的列标签。

我可以在循环中执行此操作，但我想使用矩阵函数来提高速度。

如果不编写自己的lib函数，我怎么能这样做？

Answer 1

有一个功能可以做到这一点。如果x是您的矩阵，请尝试max.col(x)。