extract_states_from_data function

Extract states for categorical variables from data