📅  最后修改于: 2023-12-03 14:39:20.402000             🧑  作者: Mango
argmin
是一个用于找到数组中最小值所在位置的函数。当数组是一个二维数组时,argmin
可以用于返回最小值所在的行和列的索引。
以下是使用 Python 中的 argmin
函数来实现的示例代码:
import numpy as np
def find_minimum_value(array):
# 将二维数组转化为 numpy 的多维数组
array = np.array(array)
# 使用 argmin 找到最小值所在的索引
min_index = np.unravel_index(np.argmin(array), array.shape)
# 返回最小值所在的行和列的索引
return min_index
# 示例二维数组
array = [[8, 2, 7],
[3, 5, 4],
[1, 9, 6]]
# 调用函数并打印结果
result = find_minimum_value(array)
print(result)
运行以上代码将输出 (2, 0)
,表示最小值在第 2 行第 0 列。
该程序使用了 numpy
库,通过 import numpy as np
将其导入,以便使用该库中的函数。首先,将二维数组转换为 numpy
的多维数组,然后使用 np.argmin
找到最小值的位置索引,最后使用 np.unravel_index
将一维索引转换为二维索引。
你可以根据自己的需求修改程序中的二维数组,并使用 find_minimum_value
函数获取最小值所在的行和列的索引。
注意:上述代码中的二维数组是一个纯 Python 的列表,如果使用 numpy
提供的数组,将更好地利用 numpy
的优势。