Skip to main content

Removing Columns and Rows

In this tutorial, we'll explore how to efficiently remove both columns and rows from Pandas DataFrames using the drop method. This method provides a convenient way to manipulate the structure of your DataFrame by eliminating unwanted columns or rows. We'll cover various scenarios, including removing columns by name, removing rows by index, and specifying the axis explicitly.

1. Importing Pandas and Reading the CSV File​

First, let's import the Pandas library and read a CSV file. We'll use a sample CSV file, but for the sake of demonstration, we'll only read the first three rows to conserve memory.

import pandas as pd

# Read CSV file, only first three rows
df = pd.read_csv('your_file.csv', nrows=3)

2. Selecting Columns​

2.1. Selecting Specific Columns​

You can select specific columns by passing a list of column names to the DataFrame.

# Select specific columns
selected_columns = ['series', 'genre']
selected_df = df[selected_columns]

2.2. Excluding Specific Columns​

To exclude specific columns, using drop method is more convenient, especially when dealing with a large number of columns.

# Remove specified columns
df.drop(['post a link', 'certificate', 'overview'], axis=1, inplace=True)

3. Removing Rows by Index​

You can also remove rows by their index using the drop method.

# Remove rows by index
df.drop([2, 3], inplace=True)

4. Specifying Axis Explicitly​

4.1. Axis for Columns​

By default, the drop method operates on columns. Explicitly specifying axis as 1 or 'columns' is redundant but can be more descriptive.

# Remove column by name
df.drop('director', axis=1, inplace=True)

4.2. Axis for Rows​

To remove rows explicitly, specify axis as 0 or 'index'.

# Remove row by index
df.drop(1, axis=0, inplace=True)

Summary​

In summary, the drop method in Pandas provides a flexible way to remove both columns and rows from DataFrames. By specifying the appropriate axis and labels, you can precisely control the operation to tailor your DataFrame according to your requirements.

Now you have the knowledge to efficiently manipulate the structure of your DataFrames in Pandas using the drop method!