how to keep only two columns in pandas code example Example: df only take 2 columns df1 = df[['a', 'b']]