我有一个matlab代码，它的目的是找到一个2维矩阵的每一列的最大值索引。
[~, indk] = max(R2)
我想在python中应用它，但似乎找不到任何好的方法。在我的例子中，变量R2是一个18x122880的矩阵。我希望indk是一个1x122880的数组。我开始这样做：
indk = np.argmax(R2)
但这只让我得到了一个索引。有没有像MATLAB代码那样的方法来处理这个问题？
oprakyz71#
就为了这个R2.argmax(axis=0) #或1，取决于所需的轴
R2.argmax(axis=0)
xwbd5t1u2#
indk = np.argmax(R2, axis=0)
axis=0表示矩阵的列。
axis=0
h6my8fg23#
numpy.ndarray.argmax
import numpy arr = numpy.array([[1,2,3],[6,5,4],[7,8,9]]) print(arr.argmax(axis = 1)) # [2 0 2] print(arr.argmax(axis = 0)) # [2 2 2]
3条答案
按热度按时间oprakyz71#
就为了这个
R2.argmax(axis=0)#或1，取决于所需的轴xwbd5t1u2#
axis=0表示矩阵的列。h6my8fg23#
numpy.ndarray.argmax