例2.5.2使用列表推导式查找数组中最大元素的所有位置

fang--- / 2024-10-18 / 原文

from numpy.random import randint
import numpy as np
a = randint(10,20,16)
ma = max(a)
ind1 = [index for index,value in enumerate(a) if value == ma]
ind2 = np.where(a==ma)
print(ind1)
print(ind2[0])
print("学号:3008")

结果如下图所示