numpy中np.max与np.argmax的区别及如何在多维数组中找到最大值索引的方法

np.max:返回数组中的最大值;
np.argmax: 返回数组中最大值坐标。
输入代码
numpy中np.max与np.argmax的区别及如何在多维数组中找到最大值索引的方法_第1张图片
运行后输出为:
numpy中np.max与np.argmax的区别及如何在多维数组中找到最大值索引的方法_第2张图片
但是在多维数组中,np.argmax方法并不适用,如下:
numpy中np.max与np.argmax的区别及如何在多维数组中找到最大值索引的方法_第3张图片
代码输出为:
numpy中np.max与np.argmax的区别及如何在多维数组中找到最大值索引的方法_第4张图片
在这里介绍一种如何在多维数组中找到最大值索引的方法:使用np.where函数
numpy中np.max与np.argmax的区别及如何在多维数组中找到最大值索引的方法_第5张图片
代码输出为:
numpy中np.max与np.argmax的区别及如何在多维数组中找到最大值索引的方法_第6张图片

你可能感兴趣的:(动态规划,线性代数,算法)