code-example code snippets in python

create a feature based on a condition for 3 possible outpus in python