NumPy: Get the index of a maximum element in a numpy array along one axis
NumPy: Array Object Exercise-120 with Solution
Write a NumPy program to get the index of a maximum element in a numpy array along one axis.
import numpy as np a = np.array([[1,2,3],[4,3,1]]) print("Original array:") print(a) i,j = np.unravel_index(a.argmax(), a.shape) print("Index of a maximum element in a numpy array along one axis:") print(a[i,j])
Original array: [[1 2 3] [4 3 1]] Index of a maximum element in a numpy array along one axis: 4
Python Code Editor:
Have another way to solve this solution? Contribute your code (and comments) through Disqus.
What is the difficulty level of this exercise?
Test your Python skills with w3resource's quiz
Python: Tips of the Day
>>> m = [x ** 2 for x in range(5)] >>> m [0, 1, 4, 9, 16]
- New Content published on w3resource:
- Scala Programming Exercises, Practice, Solution
- Python Itertools exercises
- Python Numpy exercises
- Python GeoPy Package exercises
- Python Pandas exercises
- Python nltk exercises
- Python BeautifulSoup exercises
- Form Template
- Composer - PHP Package Manager
- PHPUnit - PHP Testing
- Laravel - PHP Framework