Set Column Names when Reading CSV as pandas DataFrame in Python (Example)

 

In this article you’ll learn how to change the column names of a pandas DataFrame when importing a CSV file in the Python programming language.

The tutorial contains one example for the modification of the column names of a pandas DataFrame when importing a CSV file. More precisely, the tutorial contains the following:

Let’s dive right into the example:

 

Example Data & Software Libraries

In order to use the functions of the pandas library, we first have to import pandas to Python:

import pandas as pd                           # Import pandas library

Let’s also create some example data in Python:

data = pd.DataFrame({'x1':range(121, 127),    # Create pandas DataFrame
                     'x2':['a', 'b', 'c', 'd', 'e', 'f'],
                     'x3':range(41, 47),
                     'x4':[5, 5, 2, 9, 4, 7]})
print(data)                                   # Print pandas DataFrame

 

table 1 DataFrame set column names when reading csv as pandas dataframe python

 

As you can see based on Table 1, our example data is a DataFrame containing six rows and four variables.

Let’s write these data to a new CSV file:

data.to_csv('data.csv', index = False)        # Export pandas DataFrame to CSV

After we’ve executed the previous Python code, a new CSV file should appear in our current working directory. We’ll use this CSV file a a basement for the following example.

 

Example: Set New Column Names when Importing CSV File

The following code illustrates how to adjust the header of a pandas DataFrame that has been loaded from a CSV file.

For this task, we have to set the skiprows argument to 1, and we have to assign a list of new column names to the names argument.

Consider the Python syntax below:

data_import = pd.read_csv('data.csv',         # Read pandas DataFrame from CSV
                          skiprows = 1,
                          names = ['col1', 'col2', 'col3', 'col4'])
print(data_import)                            # Print imported pandas DataFrame

 

table 2 DataFrame set column names when reading csv as pandas dataframe python

 

As shown in Table 2, we have managed to create a new pandas DataFrame using the previous Python syntax. This DataFrame has new variables names, i.e. col1-col4 instead of x1-x4.

 

Video, Further Resources & Summary

I have recently released a video on my YouTube channel, which illustrates the Python syntax of this tutorial. You can find the video below:

 

The YouTube video will be added soon.

 

In addition, you might have a look at the other tutorials on my homepage. I have published numerous tutorials on similar topics such as lists and indices:

 

In summary: This tutorial has illustrated how to set the column names of a pandas DataFrame when importing a CSV file in the Python programming language. Let me know in the comments section below, if you have additional questions.

 

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. Required fields are marked *

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

Top