Làm thế nào để làm cho numpy.argmax trả về tất cả các lần xuất hiện tối đa?

Question

Tôi đang cố gắng tìm một hàm trả về tất cả lần xuất hiện tối đa trong một danh sách nhất định.

numpy.argmax tuy nhiên chỉ trả về lần xuất hiện đầu tiên mà nó tìm thấy. Ví dụ:

from numpy import argmax 

list = [7, 6, 5, 7, 6, 7, 6, 6, 6, 4, 5, 6] 
winner = argmax(list) 

print winner 

chỉ cung cấp chỉ mục 0. Nhưng tôi muốn nó cung cấp cho tất cả các chỉ số: 0, 3, 5.

Answer 1

Là tài liệu của np.argmax nói: "Trong trường hợp xảy ra nhiều lần giá trị tối đa, các chỉ số tương ứng với lần xuất hiện đầu tiên được trả về"., vì vậy bạn sẽ cần một chiến lược khác.

Một tùy chọn mà bạn đã được sử dụng np.argwhere kết hợp với np.amax:

>>> import numpy as np 
>>> list = [7, 6, 5, 7, 6, 7, 6, 6, 6, 4, 5, 6] 
>>> winner = np.argwhere(list == np.amax(list)) 
>>> print winner 
[[0] 
    [3] 
    [5]] 
>>> print winner.flatten().tolist() # if you want it as a list 
[0, 3, 5] 

Answer 2

đơn giản hơn nhiều ...

danh sách [danh sách == np.max (danh sách)]