Pandas: Select a row of series/dataframe by given integer index

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

Write a Pandas program to select a row of series/dataframe by given integer index.

Sample data:
Original DataFrame
col1 col2 col3
0 1 4 7
1 4 5 8
2 3 6 9
3 4 7 0
4 5 8 1
Index-2: Details
col1 col2 col3
2 3 6 9

Sample Solution :

Python Code :

import pandas as pd
import numpy as np
d = {'col1': [1, 4, 3, 4, 5], 'col2': [4, 5, 6, 7, 8], 'col3': [7, 8, 9, 0, 1]}
df = pd.DataFrame(data=d)
print("Original DataFrame")
print(df)
result = df.iloc[[2]]
print("Index-2: Details")
print(result)

Sample Output:

 Original DataFrame
   col1  col2  col3
0     1     4     7
1     4     5     8
2     3     6     9
3     4     7     0
4     5     8     1
Index-2: Details
   col1  col2  col3
2     3     6     9              

Explanation:

The above code first creates a Pandas DataFrame df with columns col1, col2, and col3 using a dictionary 'd'.

result = df.iloc[[2]] – This code selects the third row of the DataFrame using the iloc() method with index location [2] and stores it in a new DataFrame called result.

Finally print() function prints the DataFrame containing only the third row of 'df'.

Python-Pandas Code Editor:

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

Previous: Write a Pandas program to widen output display to see more columns.
Next: Write a Pandas program to replace all the NaN values with Zero's in a column of a dataframe.