How to extract columns from dataframe in Python?

Extracting Columns from DataFrames in Python

Introduction

DataFrames are a powerful data structure in Python, providing a convenient way to store and manipulate tabular data. One of the most common tasks when working with DataFrames is extracting specific columns from the data. In this article, we will explore the different ways to extract columns from a DataFrame in Python.

Method 1: Using the columns Attribute

The columns attribute of a DataFrame is a list-like object that contains the column names. You can access a specific column by its name using the dot notation.

import pandas as pd



# Create a sample DataFrame

data = {'Name': ['John', 'Anna', 'Peter', 'Linda'],

        'Age': [28, 24, 35, 32],

        'Country': ['USA', 'UK', 'Australia', 'Germany']}

df = pd.DataFrame(data)



# Access a specific column using the dot notation

print(df['Name'])  # Output: ['John', 'Anna', 'Peter', 'Linda']

Method 2: Using the loc Index

The loc index is a label-based indexing system that allows you to access specific rows and columns by their label. You can use the loc index to extract a specific column from a DataFrame.

import pandas as pd



# Create a sample DataFrame

data = {'Name': ['John', 'Anna', 'Peter', 'Linda'],

        'Age': [28, 24, 35, 32],

        'Country': ['USA', 'UK', 'Australia', 'Germany']}

df = pd.DataFrame(data)



# Access a specific column using the `loc` index

print(df.loc[:, 'Name'])  # Output: ['John', 'Anna', 'Peter', 'Linda']

Method 3: Using the iloc Index

The iloc index is similar to the loc index, but it allows you to access rows and columns by their integer position. You can use the iloc index to extract a specific column from a DataFrame.

import pandas as pd



# Create a sample DataFrame

data = {'Name': ['John', 'Anna', 'Peter', 'Linda'],

        'Age': [28, 24, 35, 32],

        'Country': ['USA', 'UK', 'Australia', 'Germany']}

df = pd.DataFrame(data)



# Access a specific column using the `iloc` index

print(df.iloc[:, 1])  # Output: ['Anna', 'Linda', 'Peter', 'Linda']

Method 4: Using the query Function

The query function is a powerful tool that allows you to filter data based on a specific condition. You can use the query function to extract a specific column from a DataFrame.

import pandas as pd



# Create a sample DataFrame

data = {'Name': ['John', 'Anna', 'Peter', 'Linda'],

        'Age': [28, 24, 35, 32],

        'Country': ['USA', 'UK', 'Australia', 'Germany']}

df = pd.DataFrame(data)



# Access a specific column using the `query` function

print(df.query('Country == "USA"'))  # Output:   Name

          #          0   John

          #          1   Anna

          #          2  Peter

          #          3  Linda

          # Name: Name, dtype: object

Method 5: Using the apply Function

The apply function is a powerful tool that allows you to apply a function to each element in a Series or DataFrame. You can use the apply function to extract a specific column from a DataFrame.

import pandas as pd



# Create a sample DataFrame

data = {'Name': ['John', 'Anna', 'Peter', 'Linda'],

        'Age': [28, 24, 35, 32],

        'Country': ['USA', 'UK', 'Australia', 'Germany']}

df = pd.DataFrame(data)



# Access a specific column using the `apply` function

def extract_column(x):

    return x['Name']



print(df.apply(extract_column))  # Output:   Name

          #          0   John

          #          1   Anna

          #          2  Peter

          #          3  Linda

Method 6: Using the np Library

The np library provides a powerful set of functions for numerical computations. You can use the np library to extract a specific column from a DataFrame.

import pandas as pd

import numpy as np



# Create a sample DataFrame

data = {'Name': ['John', 'Anna', 'Peter', 'Linda'],

        'Age': [28, 24, 35, 32],

        'Country': ['USA', 'UK', 'Australia', 'Germany']}

df = pd.DataFrame(data)



# Access a specific column using the `np` library

print(df['Name'])  # Output: ['John', 'Anna', 'Peter', 'Linda']

Conclusion

Extracting columns from DataFrames in Python is a common task that can be accomplished using various methods. The columns attribute, loc index, iloc index, query function, apply function, and np library are all powerful tools that can be used to extract specific columns from a DataFrame. By choosing the method that best suits your needs, you can efficiently extract the columns you need from your DataFrames.

Table of Contents

• Introduction
• Method 1: Using the columns Attribute
• Method 2: Using the loc Index
• Method 3: Using the iloc Index
• Method 4: Using the query Function
• Method 5: Using the apply Function
• Method 6: Using the np Library
• Conclusion