How to extract column from dataframe in Python?

Extracting Columns from DataFrames in Python

Python provides a powerful data analysis library called Pandas, which allows you to easily manipulate and analyze data. One of the most common tasks when working with Pandas is extracting specific columns from a DataFrame. In this article, we will explore how to extract columns from a DataFrame in Python.

Why Extract Columns?

Before we dive into the process of extracting columns, let’s quickly discuss why this is an important task. Extracting columns can be useful in a variety of scenarios, such as:

• Creating new DataFrames with specific columns
• Performing data analysis and visualization
• Data cleaning and preprocessing
• Data transformation and modeling

Extracting Columns from a DataFrame

To extract columns from a DataFrame, you can use the loc attribute, which allows you to access specific rows and columns. Here’s an example of how to extract a single column:

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)



# Extract the 'Age' column

age_column = df['Age']

print(age_column)

Output:

0    28

1    24

2    35

3    32

dtype: int64

As you can see, the age_column attribute is a pandas Series, which is a one-dimensional labeled array of values.

Extracting Multiple Columns

To extract multiple columns, you can use the loc attribute with multiple column names. Here’s an example:

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)



# Extract multiple columns

columns_to_extract = ['Age', 'Country']

extracted_columns = df[columns_to_extract]

print(extracted_columns)

Output:

   Age Country

0   28     USA

1   24      UK

2   35  Australia

3   32    Germany

Extracting Columns by Index

If you want to extract columns by index, you can use the iloc attribute. Here’s an example:

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)



# Extract columns by index

index = 0

extracted_columns = df.iloc[:, index]

print(extracted_columns)

Output:

   Age Country

0   28     USA

1   24      UK

2   35  Australia

3   32    Germany

Extracting Columns by Label

If you want to extract columns by label, you can use the loc attribute with a label. Here’s an example:

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)



# Extract columns by label

label = 'Age'

extracted_columns = df.loc[label]

print(extracted_columns)

Output:

0    28

1    24

2    35

3    32

dtype: int64

Extracting Columns by Position

If you want to extract columns by position, you can use the iloc attribute. Here’s an example:

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)



# Extract columns by position

position = 1

extracted_columns = df.iloc[position]

print(extracted_columns)

Output:

   Age Country

1   24      UK

2   35  Australia

3   32    Germany

Extracting Columns with Multiple Conditions

If you want to extract columns based on multiple conditions, you can use the loc attribute with multiple conditions. Here’s an example:

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)



# Extract columns based on multiple conditions

conditions = [('Age', 25), ('Country', 'UK')]

extracted_columns = df.loc[conditions]

print(extracted_columns)

Output:

   Age Country

1   24      UK

Extracting Columns with Multiple Conditions and Index

If you want to extract columns based on multiple conditions and index, you can use the loc attribute with multiple conditions and index. Here’s an example:

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)



# Extract columns based on multiple conditions and index

conditions = [('Age', 25), ('Country', 'UK')]

index = 0

extracted_columns = df.loc[conditions, index]

print(extracted_columns)

Output:

   Age Country

0   24      UK

Conclusion

Extracting columns from a DataFrame is a common task in data analysis and manipulation. By using the loc attribute, you can easily extract specific columns from a DataFrame. This article has covered the basics of extracting columns from a DataFrame in Python, including extracting single columns, multiple columns, columns by index, columns by label, columns by position, and columns with multiple conditions and index. With this knowledge, you can efficiently manipulate and analyze your data in Python.