按变量切片多维数组

Question

我正在编写一个方法，它接受一个二维数组和一个 int 行号作为参数，并返回给定行中元素的最大值。

看起来像这样：

function getHighestInRow(A, i)
        return(maximum(A[:i,:]))
end

我遇到的问题是当我用

切片数组时

A[:i,:]

我收到一个参数错误，因为 :i 使 i 得到不同的处理。

代码在另一个方向上工作

A[:,i,:]

有没有办法逃脱冒号？以便i 在冒号后被视为变量？

Answer 1

你对冒号做了一些奇怪的事情。在这种情况下，您使用的是符号 :i 而不是 i 的值。只需 getHighestInRow(A,i) = maximum(A[i,:]) 即可。

编辑：正如 Dan Getz 在对该问题的评论中所说，getHighestInRow(A,i) = maximum(@view A[i,:]) 效率更高，因为切片将分配一个临时的不必要的数组。