take only two columns from dataframe pandas code example Example: df only take 2 columns df1 = df[['a', 'b']]