DF принимают только 2 столбца

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