📜  argmin 为二维数组返回一个值 - Python (1)

📅  最后修改于: 2023-12-03 14:39:20.402000             🧑  作者: Mango

argmin 为二维数组返回一个值 - Python

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 的优势。