import numpy as np
df = pd.DataFrame({'Type': list('ABBC'), 'Set': list('ZZXY')})
df['color'] = np.where(df['Set'] == 'Z', 'green', 'red')
print(df)
Условное создание Pandas столбца серии / фрейма данных