Get Index of Rows with Match in Column in Python (Example)

 

In this Python tutorial you’ll learn how to return the indices of rows with a certain value.

Table of contents:

If you want to know more about these contents, keep reading…

 

Example Data & Libraries

We first have to import the pandas library.

import pandas as pd                                         # Import pandas library

Furthermore, have a look at the example data below:

data = pd.DataFrame({'x1':['a', 'b', 'c', 'd', 'e', 'f'],  # Create example DataFrame
                     'x2':[5, 2, 5, 5, 8, 1],
                     'x3':['A', 'B', 'C', 'D', 'E', 'F']})
print(data)                                                # Print example DataFrame

 

table 1 DataFrame get index rows match column python

 

Table 1 shows the structure of our example data – It is composed of six rows and the three columns “x1”, “x2”, and “x3”.

 

Example: Return Matching Row Indices Using index Attribute & tolist() Function

In this example, I’ll illustrate how to find the indices of all rows where the column x2 contains the value 5.

For this, we can use the index attribute of our pandas DataFrame in combination with the tolist function.

Have a look at the following Python code:

data.index[data['x2'] == 5].tolist()                       # Get row indices
# [0, 2, 3]

As you can see, a list containing the values 0, 2, and 3 has been returned to the console. These values are the row indices where the variable x2 matches the value 5.

 

Video, Further Resources & Summary

Have a look at the following video on the techtipnow YouTube channel. The speaker of this video demonstrates how to use the tolist function and related commands in the Python programming language.

 

Please accept YouTube cookies to play this video. By accepting you will be accessing content from YouTube, a service provided by an external third party.

YouTube Content Consent Button Thumbnail

YouTube privacy policy

If you accept this notice, your choice will be saved and the page will refresh.

 

Also, you may want to read the related tutorials on this homepage.

 

In this Python tutorial you have learned how to return the indices of rows with a certain value in columns of a pandas DataFrame. In case you have further questions, don’t hesitate to let me know in the comments section. Furthermore, don’t forget to subscribe to my email newsletter for updates on new articles.

 

Subscribe to the Statistics Globe Newsletter

Get regular updates on the latest tutorials, offers & news at Statistics Globe.
I hate spam & you may opt out anytime: Privacy Policy.


Leave a Reply

Your email address will not be published.

Fill out this field
Fill out this field
Please enter a valid email address.

Menu
Top