/Python/Argmax change dafault value for multiple maxima

Argmax change dafault value for multiple maxima

def my_argmax(a):
    rows = np.where(a == a.max(axis=1)[:, None])[0]
    rows_multiple_max = rows[:-1][rows[:-1] == rows[1:]]
    my_argmax = a.argmax(axis=1)
    my_argmax[rows_multiple_max] = -1
    return my_argmax

Source: stackoverflow.com

Similar codes

2 votes347 views2 answers
Numpy moving average
Python
-2 votes333 views2 answers
Savefig matplotlib python
Python
1 vote333 views1 answer
Upload_file boto3 headers
Python
0 vote355 views1 answer
Gspread_pandas pypi
Python
44 votes355 views6 answers
Plt figsize
Python
0 vote353 views1 answer
Recursive python program to print numbers from n to 1
Python
9 votes367 views1 answer
Drop last row pandas
Python
2 votes374 views4 answers
Python slow print
Python
12 votes324 views2 answers
Python read from stdin
Python
0 vote346 views1 answer
Log transform pandas dataframe
Python