select column from dataframe in python code example Example: df only take 2 columns df1 = df[['a', 'b']]