Python DataFrame – Assign New Labels to Columns

Python Dataframe Columns - Labels assigned new value

In this post, you will get a code sample related to how to assign new labels to columns in python programming while training machine learning models. 

This is going to be very helpful when working with classification machine learning problem. Many a time the labels for response or dependent variable are in text format and all one wants is to assign a number such as 0, 1, 2 etc instead of text labels. Beginner-level data scientists will find this code very handy.

We will look at the code for the dataset as represented in the diagram below:

Python - Read CSV File

Fig 1. Python – Read CSV File

In the above code, you will see that class labels are named as very_low, Low, High, Middle etc. The objective is to provide label as 1, 2, 3, 4 etc.

Code for Assigning New Class Labels

Here are the steps:

Define a function which returns different numeric value based on the label

def assignNewLabels(label):
    if label == 'very_low' or label == 'Very Low':
        return 1
    elif label == 'Low':
        return 2
    elif label == 'Middle':
        return 3
    elif label == 'High':
        return 4
        return -1

Next step is to use apply method

df1['UNS_NEW'] = df1['UNS'].apply(assignNewLabels)

The above code creates a new column namely UNS_NEW. The data frame looks like the following:

Python Data frame columns - Labels assigned new value

Fig 2. Python Data frame columns – Labels assigned new value

Note that there is a new column with new labels (1, 2, 3, 4) assigned based on the logic within assignNewLabels function.

Ajitesh Kumar
Follow me

Ajitesh Kumar

I have been recently working in the area of Data Science and Machine Learning / Deep Learning. In addition, I am also passionate about various different technologies including programming languages such as Java/JEE, Javascript, Python, R, Julia etc and technologies such as Blockchain, mobile computing, cloud-native technologies, application security, cloud computing platforms, big data etc. I would love to connect with you on Linkedin.
Posted in AI, Data Science, Machine Learning, News. Tagged with , , .

Leave a Reply

Your email address will not be published. Required fields are marked *

Time limit is exhausted. Please reload the CAPTCHA.