This is easy to implement with standard Python libraries ... It’s also important to do your exploratory data analysis (step four) before modeling, to avoid introducing biases in your predictions.