How to make numpy.argmax return all occurrences of the maximum?

by

As documentation of np.argmax says: “In case of multiple occurrences of the maximum values, the indices corresponding to the first occurrence are returned.”, so you will need another strategy.

One option you have is using np.argwhere in combination with np.amax:

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

More Related Contents:

  1. How do I get indices of N maximum values in a NumPy array?
  2. How to get the index of a maximum element in a NumPy array along one axis
  3. numpy max vs amax vs maximum
  4. Max in a sliding window in NumPy array
  5. How to calculate rolling / moving average using python + NumPy / SciPy?
  6. Intuition and idea behind reshaping 4D array to 2D array in NumPy
  7. Removing nan values from an array
  8. Python: Maximum recursion depth exceeded
  9. How to count the occurrence of certain item in an ndarray?
  10. How do I compute derivative using Numpy?
  11. binning data in python with scipy/numpy
  12. How do I count the occurrence of a certain item in an ndarray?
  13. Calculating Pearson correlation and significance in Python
  14. ValueError: could not broadcast input array from shape (224,224,3) into shape (224,224)
  15. Sliding window of M-by-N shape numpy.ndarray
  16. numpy array concatenate: “ValueError: all the input arrays must have same number of dimensions”
  17. Precision lost while using read_csv in pandas
  18. How to split a numpy array in fixed size chunks with and without overlap?
  19. reading v 7.3 mat file in python
  20. Pandas in AWS lambda gives numpy error
  21. Pandas groupby with categories with redundant nan
  22. Shuffling NumPy array along a given axis
  23. How to display progress of scipy.optimize function?
  24. Easy way to test if each element in an numpy array lies between two values?
  25. Python numpy: cannot convert datetime64[ns] to datetime64[D] (to use with Numba)
  26. ImportError: No module named numpy on spark workers
  27. What is the most efficient way to check if a value exists in a NumPy array?
  28. ImportError: DLL load failed when importing Numpy installed in conda virtual environment
  29. Fast Way to slice image into overlapping patches and merge patches to image
  30. How can I make a numpy ndarray from bytes?

Leave a Comment