Pandas conditional collumn

Make a condition statement on column pandas

df['color'] = ['red' if x == 'Z' else 'green' for x in df['Set']]

Source: stackoverflow.com

Pandas conditional collumn

import pandas as pd
import numpy as np

df = pd.DataFrame({'Type':list('ABBC'), 'Set':list('ZZXY')})
df['color'] = np.where(df['Set']=='Z', 'green', 'red')
print(df)

Source: stackoverflow.com

Similar codes

1 vote330 views2 answers
Tensor vs numpy array
Python
0 vote316 views1 answer
How to write flow of execution in python
Python
-1 vote342 views1 answer
The field was declared on serializer, but has not been included in the 'fields' option
Python
0 vote355 views1 answer
Numpy combinations of 5 bits
Python
0 vote333 views1 answer
(function(a_,%20b_)%20%7B%20with%20(a_)%20with%20(b_)%20return%20summary%20%7D)
Python
0 vote307 views1 answer
Python update flask
Python
2 votes308 views1 answer
Two dimensional array python
Python
0 vote309 views1 answer
Why we need open ( ) and close ( ) in os
Python
5 votes296 views6 answers
How to check a string in if statement python
Python
0 vote351 views1 answer
Lambda Functions using for loop
Python