Selecting Columns in Pandas

Column Selection Techniques in Pandas

Column Selection Techniques in Pandas

This blog post aims to explore different methods for selecting columns in Pandas DataFrames, inspired by Matt Harrison's book, Effective Pandas. The Python library Pandas provides multiple flexible and efficient ways to manipulate, analyze, and visualize data. One of the most common tasks in data wrangling is column selection. Let's examine some effective techniques for this.

Initial Setup


import pandas as pd
loc = 'https://raw.githubusercontent.com/aew5044/Python---Public/main/movie.csv'
m = pd.read_csv(loc)
m.head()

Basic Column Selection

You can start by selecting columns directly by their names.


ma = m[['actor_1_name', 'actor_2_name', 'actor_3_name', 'director_name', 'movie_title','gross']]
ma.head()

Using Regex for Column Selection

Regular expressions (Regex) can be powerful tools for pattern matching. The filter method can accept regex patterns.


m.filter(regex='^actor|director')

Rename Columns

To make your DataFrame more manageable, you can rename column names. Here, a function named shorten is used to rename columns.


def shorten(col):
    return (
        col.replace('facebook_likes', 'fb')
        .replace('_for_reviews', '')
        .replace('_name', '')
    )
ma = m.rename(columns=shorten)
ma.head()

Selection by Data Types

You can also select columns based on their data types using the select_dtypes method.


m.select_dtypes(include='number').head()
m.select_dtypes(include=['object','int']).head()
m.select_dtypes(exclude='float').head()

Substring Matching

The filter method can be used to match a substring within the column names.


m.filter(like='num').head()

Combining Multiple Filters

You can combine multiple criteria to filter out columns using Regex and the filter method.


m.filter(regex='^actor|director|movie_title|gross').head()

Conclusion

Effective column selection in Pandas is crucial for data manipulation and analysis. The methods outlined in this post offer a range of options for different scenarios. Understanding how to combine these tools can make your data wrangling processes more efficient and robust.

Link to Google Colab

Enjoy the Notebook - link

Comments

Post a Comment

Popular posts from this blog

Drawing Tables with ReportLab: A Comprehensive Example

Image
Tables play an integral part in visualizing and summarizing complex data in a clear, readable format. ReportLab offers a powerful object called `Table` for rendering data in table form within a PDF. By pairing this with `TableStyle`, users can customize tables to fit any styling requirements. Let's break down an example to understand the power and flexibility of ReportLab's table creation capabilities. The first step is to know your table and cast it to a list of list.  I didn't know how to do this off the top of my head, so some quick searching gave me the results! It is a common task for ReportLab users so there are many example out there.  Here is my example:  #Start working with the table budgetinlist = [budget.columns.tolist()] + budget.values.tolist() # Format numbers with commas for i in range(1, len(budgetinlist)):     for j in range(1, len(budgetinlist[i])):         try:             budgetinlist[i][j] = ...
Read more

Blog Topics

Image
This platform aims to systematically curate and preserve my favored Python techniques for future reference. This digital journal, accessible globally wherever an internet connection is available, is a convenient and effective learning tool.  Going down a wooded path at night, alone, with your trusted companion by your side, can be scary and daunting.  However, with support from your friends, documenting your path (curating a code base like a blog), and continuously working on the code base can make the process appear as a walk in the park!  In a world dominated by sophisticated language models like OpenAI's GPT, blogging may seem like a relic of the past. But here's a nuanced perspective: this blog serves as an innovative instrument for both teaching and mastering new programming languages. It's about immersing in syntax, applying it, and then translating the information into layman's terms for broader understanding. Each topic presented here is elaborated in a separate...
Read more

fillna

Image
Handling Missing Data with Pandas: A Comprehensive Guide Handling Missing Data with Pandas: A Comprehensive Guide Dealing with missing data is an essential part of the data cleaning process in Python programming. The Pandas library provides various methods to fill or drop missing values, depending on the nature of the data and the desired outcome. In this guide, we'll explore different techniques to handle missing data, including hard-coded values, filling specific columns, forward fill, and using rolling averages. 1. Filling All Missing Values with a Fixed Number You can simply use the fillna() function to replace all NaNs with a specific value: import pandas as pd df = pd.DataFrame({'A': [1, 2, None], 'B': [None, 5, 6]}) df.fillna(0) 2. Filling Specific Columns with Different Values df.fillna({1: 0.5, 2: 0}) 3. Forward Fill Method df.fillna(method='ff...
Read more