numpy ndarray求其最值的索引

import numpy as np

a = 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]