Python loc() function – Extract values from a dataset

Python Loc() Function

Hey readers! In this article, we will be focusing on the functioning of Python loc() function in detail. So, let us get started!!

Working of Python loc() function

Python comprises various modules that have in-built functions to deal with and manipulate the data values.

One such module is Pandas module.

Pandas module enables us to handle large data sets containing a considerably huge amount of data for processing altogether.

This is when Python loc() function comes into the picture. The loc() function helps us to retrieve data values from a dataset at an ease.

Using the loc() function, we can access the data values fitted in the particular row or column based on the index value passed to the function.


pandas.DataFrame.loc[index label]

We need to provide the index values for which we want the entire data to be represented in the output.

The index label may be one of the below values:

  • Single label – example: String
  • List of string
  • Slice objects with labels
  • List of an array of labels, etc.

Thus, we can retrieve a particular record from a dataset based upon the index label using the loc() function.

Note: If the passed index is not present as a label, it returns KeyError.

Let us now focus on the implementation of the same using the below examples.

Examples of Python loc() function

Let us first create a data frame with a set of data values using data frame in the Pandas module as shown below:

import pandas as pd
data = pd.DataFrame([[1,1,1], [4,4,4], [7,7,7], [10,10,10]],
     index=['Python', 'Java', 'C','Kotlin'],


Python	1	1	1
Java	4	4	4
C	7	7	7
Kotlin	10	10	10

Having created the data frame with a defined set of values, let us now try to retrieve a set of rows or columns having data values for a particular index as shown below:

Extract One Row from a Data frame


So, using the above command, we have extracted all the data values associated with the index label ‘Python’.


RATE    1
EE      1
AA      1
Name: Python, dtype: int64

Extract Multiple Rows from a Data frame

Let us now try to extract the data rows and columns associated with multiple indexes at the same time using the below command.



          RATE  EE  AA
Python     1    1    1
C          7    7    7

Extract Range of Rows using Python loc()


Here, we have used the slice object as with labels to display the rows and columns associated with the labels from ‘Python’ to ‘C’.


          RATE  EE  AA
Python     1   1   1
Java       4   4   4
C          7   7   7


By this, we have come to the end of this topic. Feel free to comment below, in case you come across any question.

For more such posts related to Python, Stay tune and till then Happy Learning!!