how to extract columns from dataframe in python code example

Example 1: python code to drop columns from dataframe

# Let df be a dataframe
# Let new_df be a dataframe after dropping a column

new_df = df.drop(labels='column_name', axis=1)

# Or if you don't want to change the name of the dataframe
df = df.drop(labels='column_name', axis=1)

# Or to remove several columns
df = df.drop(['list_of_column_names'], axis=1)

# axis=0 for 'rows' and axis=1 for columns

Example 2: python extract specific columns from pandas dataframe

# Basic syntax:
new_dataframe = dataframe.filter(['col_name_1', 'col_name_2'])
# Where the new_dataframe will only have the column names specified

# Note, use df.filter(['names', ... ], axis=0] to select rows

Example 3: python - subset specific columns name in a dataframe

columns = ['b', 'c']
df1 = pd.DataFrame(df, columns=columns)

Example 4: pandas iloc select certain columns

dataframe.iloc[:,[1,2]]

Example 5: remove columns from a dataframe python

df = df.drop(df.columns[[0, 1, 3]], axis=1)  # df.columns is zero-based pd.Index

Example 6: extract one column from dataframe python

import pandas as pd

input_file = "C:\\....\\consumer_complaints.csv"
dataset = pd.read_csv(input_file)
df = pd.DataFrame(dataset)
cols = [1,2,3,4]
df = df[df.columns[cols]]