/Python/Finding the maximum element from a matrix with Python numpy.argmax()

Finding the maximum element from a matrix with Python numpy.argmax()

import numpy as np    

a = np.matrix([[1,2,3,33],[4,5,6,66],[7,8,9,99]])

print(np.argmax(a))  # 11, which is the position of 99
print(np.argmax(a[:,:]))  # 11, which is the position of 99
print(np.argmax(a[:1]))  # 3, which is the position of 33
print(np.argmax(a[:,2]))  # 2, which is the position of 9
print(np.argmax(a[1:,2]))  # 1, which is the position of 9

Source: itsmycode.com

Similar codes

97 votes322 views6 answers
How to read a csv file in python
Python
5 votes360 views1 answer
Sphere volume formula
Python
16 votes345 views6 answers
Python turtle setup
Python
0 vote342 views1 answer
Xgboost feature importance
Python
2 votes349 views2 answers
If else pandas dataframe
Python
5 votes334 views6 answers
Pytube python
Python
8 votes346 views2 answers
How to launch an application using python
Python
28 votes396 views6 answers
Convert list to string python
Python
5 votes346 views4 answers
Dropna for specific column pandas
Python
36 votes319 views6 answers
Get list file in folder python
Python