select only 2 columns from dataframe code example

Example: df only take 2 columns

df1 = df[['a', 'b']]