Pandas conditional creation of a series/dataframe column
Asked 07 September, 2021
Viewed 2.5K times
  • 53
Votes

I have a dataframe along the lines of the below:

    Type       Set
1    A          Z
2    B          Z           
3    B          X
4    C          Y

I want to add another column to the dataframe (or generate a series) of the same length as the dataframe (equal number of records/rows) which sets a colour 'green' if Set == 'Z' and 'red' if Set equals anything else.

What's the best way to do this?

10 Answer