julia - 使用 Julia 1.0 findmax 等效于 numpy.argmax

Question

在 Julia 中，我想为每行中的最大值找到矩阵的列索引，结果为 Vector{Int} .这是我目前的做法(Samples 有 7 列和 10,000 行):

mxindices = [ i[2] for i in findmax(Samples, dims = 2)[2]][:,1]

这有效，但感觉相当笨拙和冗长。想知道有没有更好的办法。

Answer 1

更简单:Julia 有一个 argmax函数和 Julia 1.1+ 有一个 eachrow迭代器。因此:

map(argmax, eachrow(x))

简单、可读、快速——它与 Colin 的 f3 的性能相匹配和 f4在我的快速测试中。