追踪布尔掩码的NumPy数组中argmin/argmax的原始位置 - Python

下面是一个示例代码，展示了如何追踪布尔掩码的NumPy数组中argmin/argmax的原始位置：

import numpy as np

# 创建一个示例numpy数组
arr = np.array([1, 2, 3, 4, 5])

# 使用argmin找到最小值的索引
min_index = np.argmin(arr)

# 创建一个布尔掩码，表示arr中等于最小值的位置
mask = (arr == arr[min_index])

# 使用np.where函数找到布尔掩码中为True的位置
min_positions = np.where(mask)[0]

print("最小值的位置：", min_positions)

输出结果为：

最小值的位置： [0]

同样的，你也可以用argmax来找到最大值的位置。将np.argmin替换为np.argmax，即可找到最大值的位置。

希望对你有所帮助！