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

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")

结果如下图所示

posted @ 2024-10-15 16:38  方~~  阅读(18)  评论(0)    收藏  举报