How can I build customized regression models for different categories in the same python dataframe?

Question

I have a data frame in python with the following features:

• country
• age
• weight
• height

I want to predict the "height