DF weź tylko 2 kolumny

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