I have a pandas dataframe that looks like this:
COLOR SHAPE 0 red square 1 red oval 2 blue circle 3 green oval 4 red square 5 purpl