python argmin_Python中的np.argmin()和np.argmax()

按照axis的要求返回最小的数/最大的数的下标

numpy.argmin(a, axis=None, out=None) numpy.argmax(a, axis=None, out=None)

a ：传入一个数组， axis： 默认将输入数组展平，否则，按照axis方向 out：可选

import numpy as np

a = np.arange(6).reshape(2, 3)

a

array([[0, 1, 2],

[3, 4, 5]])

# 此时自动展平了

np.argmin(a)

0

np.argmax(a)

5

# 在axis方向上找最小的值并返回坐标

np.argmin(a, axis=1)

array([0, 0], dtype=int64)

# 在axis方向上找最大的值并返回坐标

np.argmax(a, axis=1)

array([2, 2], dtype=int64)

当有多个最小值的时候只显示第一个

a = np.arange(6)

a[4] = 0

a

array([0, 1, 2, 3, 0, 5])

np.argmin(a)

0