drop columns that are not in train dataframe python code example

Example 1: drop a column from dataframe

#working with "text" syntax for the columns:
df.drop(['column_nameA', 'column_nameB'], axis=1, inplace=True)

Example 2: drop variable pandas

# pandas drop a column with drop function
gapminder_ocean.drop(['pop'], axis=1)