how to extract specific columns from dataframe in python code example
Example 1: python extract specific columns from pandas dataframe
new_dataframe = dataframe.filter(['col_name_1', 'col_name_2'])
Example 2: python: select specific columns in a data frame
df = df[["Column_Name1", "Column_Name2"]]
Example 3: 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]]