get columns needed for model python code example Example: selecting items in a column of a dataframe df.loc[df['column_name'] == some_value]