python如何返回一个数组中不为0的元素的最大索引值

• 使用 numpy.argmax 函数结合布尔索引来找到非零元素中的最大值索引。

numpy.argmax() 应用

• 非零元素最大索引：首先通过布尔索引 arr != 0 筛选出数组 arr 中的非零元素，然后使用 numpy.argmax 函数找到这些非零元素中的最大值索引。10

示例代码如下：

import numpy as np

arr = np.array([0, 2, 3, 0, 5, 0, 7, 0])
nonzero_arr = arr[arr != 0]  # 取出非零元素
max_index = np.argmax(nonzero_arr)  # 计算非零元素中的最大值索引
print(max_index)  # 输出最大值索引

这样，max_index 将包含非零元素中最大值的索引。10