正文

numpy ndarray求其最值的索引

IP云
IP云V管理员/4 阅读
07/31
提示：扫一扫查出行
【扫一扫了解最新限行尾号】
复制提示

import numpy as npa = np.array([1,2,3,4])
np.where(a== np.max(a))
>>>3

但假设其最值不止一个，如下

a = np.array([1,2,3,4,4])
np.where(a== np.max(a))
>>>[3,4]