[ 常見問題 ] find the max from each row in python

 Source From Here

Question

view plaincopy to clipboardprint?

1. 0.511474    0.488526  
2. 0.468783    0.531217  
3. 0.35111     0.64889  
4. 0.594834    0.405166  

"How to find the max from each row in python and store it in an numpy array or pandas Dataframe and store it in an numpy array , i.e the output below"

HowTo
Use the numpy amax function. np.amax:

view plaincopy to clipboardprint?

1. import numpy as np  
2. a = np.array([[0.511474,    0.488526],  
3.             [0.468783,    0.531217],  
4.             [0.35111,     0.64889],  
5.             [0.594834,    0.405166]])  
6. x = np.amax(a, 1)  
7. print(x)  

which returns:

[0.511474 0.531217 0.64889 0.594834]