take only specific column pandas 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: how to keep only certain columns in dataframe

# Using DataFrame.drop
df.drop(df.columns[[1, 2]], axis=1, inplace=True)

# drop by Name
df1 = df1.drop(['B', 'C'], axis=1)

# Select the ones you want
df1 = df[['a','d']]