dataframe extract column code example

Example 1: 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 2: python - subset specific columns name in a dataframe

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

Example 3: how to pick out separate columns from the pandas dataframe object

df1 = df.iloc[:, 0:2] # If you want to do it by index. Remember that Python does not slice inclusive of the ending index.
df1 = df[['a', 'b']] ## if you want to do it b nae

Example 4: python: select specific columns in a data frame

df = df[["Column_Name1", "Column_Name2"]]

Example 5: 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]]