Python中的 numpy.argmin()

简介

numpy.argmin()函数返回数组中最小值的索引。如果数组有多个最小值，则返回第一个最小值的索引。

语法

numpy.argmin(arr, axis=None, out=None)

• arr：输入数组。
• axis：沿着这个轴比较，如果为None则比较整个数组，默认为None。
• out：输出数组，这里指定输出结果的位置。

返回值

返回输入数组中最小值的索引。

例子

import numpy as np

# 一维数组
arr = np.array([3, 4, 1, 5, 2, 6])
print(np.argmin(arr))  # 输出 2

# 多维数组
arr = np.array([[3, 4], [1, 5], [2, 6]])
print(np.argmin(arr))  # 输出 2

# 沿着某个轴比较
arr = np.array([[3, 4], [1, 5], [2, 6]])
# 比较每行的最小值
print(np.argmin(arr, axis=1))  # 输出 [0 0 0]
# 比较每列的最小值
print(np.argmin(arr, axis=0))  # 输出 [1 0]

# 输出结果到指定位置
out_arr = np.zeros(3)
print(np.argmin(arr, axis=1, out=out_arr))  # 输出 [0 0 0]
print(out_arr)  # 输出 [0. 0. 0.]

参考文献

1. numpy.argmin()