return specific columns of df code example
Example 1: 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']]
Example 2: python: select specific columns in a data frame
df = df[["Column_Name1", "Column_Name2"]]