Pandas: Find the row for where the value of a given column is maximum

Last update on December 21 2024 07:42:08 (UTC/GMT +8 hours)

Write a Pandas program to find the row for where the value of a given column is maximum.

Sample Solution:

Python Code :

import pandas as pd
d = {'col1': [1, 2, 3, 4, 7], 'col2': [4, 5, 6, 9, 5], 'col3': [7, 8, 12, 1, 11]}
df = pd.DataFrame(data=d)
print("Original DataFrame")
print(df)
print("Row where col1 has maximum value:")
print(df['col1'].argmax())
print("Row where col2 has maximum value:")
print(df['col2'].argmax())
print("Row where col3 has maximum value:")
print(df['col3'].argmax())

Sample Output:

    Original DataFrame
   col1  col2  col3
0     1     4     7
1     2     5     8
2     3     6    12
3     4     9     1
4     7     5    11
Row where col1 has maximum value:
4
Row where col2 has maximum value:
3
Row where col3 has maximum value:
2              

Explanation:

The above code creates a pandas DataFrame 'df' with three columns - 'col1', 'col2', and 'col3'. The code then uses the 'argmax()' function to find the index of the maximum value in each column.

Therefore -

• The first 'print' statement returns the index of the row that has the maximum value in 'col1'.
• The second 'print' statement returns the index of the row that has the maximum value in 'col2'.
• The third 'print' statement returns the index of the row that has the maximum value in 'col3'.

Python-Pandas Code Editor:

Have another way to solve this solution? Contribute your code (and comments) through Disqus.

Previous: Write a Pandas program to create a DataFrame from a Numpy array and specify the index column and column headers.
Next: Write a Pandas program to check whether a given column is present in a DataFrame or not.