考虑矩阵:
M = [1.0 2.0 3.0; 5.0 4.0 3.0; 1.0 100.0 12.0]
我想获得每行中最大值的列。 所以它应该是:
col = [3; 1; 2]
,因为
M[1,3] -> 3.0;
M[2,1] -> 5.0;
M[3,2] -> 100.00;
在Octave中很容易实现:
[max, col] = max(M,[],2)
,其中col = [3; 2; 1]。
在Julia中,我只能找到findmax
函数,它返回每行最大元素的绝对索引。
所以它会是:
max, aindx = findmax(M,2)
,其中aindx = [7,2,6]
M [7] = 3.0; M [2] = 5.0; M [6] = 100;
哪里可以找到Octave max(M,[],2)的Julia equivelent?
我目前的解决方法:
max, aindx = findmax(M, 2);
msize=size(M);
col = zeros(msize[1], 1);
for i=1:msize[1]
_, col[i] = ind2sub(msize,aindx[i]);
end
答案 0 :(得分:5)
Julia的findmax
比Octave的max
更灵活:您可以一次找到多个维度的最大值。因此,它返回一个线性索引。
正如您所指出的,您可以使用ind2sub
来计算您想要的任何索引。如果您经常使用它,您可能希望将“变通方法”定义为一个函数,以方便使用。如果你想确保它始终可用,你可以将该功能放在.juliarc.jl
中。