python code to select columns from dataframe