Python If Else Segmentation 'Invalid Type Comparison' Error in Dataframe

Question

I have a sample dataframe (df_merged_1) shown below:

The Revenue column is a float64 dtype. I want to create a new column called 'Revenue_Segment'. This is what I want the end result to look like:

Below is the code I used to segment:

if df_merged_1['Revenue'] >= 0 and df_merged_1['Revenue'] <= 2200:
    df_merged_1['AUM_Segment'] == 'test1'
else:
    df_merged_1['AUM_Segment'] == 'test0'

But the code is not working ... I get the following error:

TypeError: invalid type comparison

Any help is greatly appreciated!

score 0 · Accepted Answer · answered Apr 23 '17 at 15:15

0

Not elegant, but this is the only solution I can think of now:

# Make a new list to add it later to your df
rev_segment = []

# Loop through the Revenue column
for revenue in df_merged_1['Revenue']:
    if revenue >= 0 and revenue <= 2200:
        rev_segment.append('test1')
    else:
        rev_segment.append('test0')

# Now append the list to a new column
df_merged_1['Revenue_Segment'] = rev_segment

answered Apr 23 '17 at 15:15

Alex

3,958
4
17
24

Brilliant! This worked perfectly ...thank you so much :) – PineNuts0 Apr 23 '17 at 17:45

score 0 · Answer 2 · answered Apr 23 '17 at 15:21

0

Try this:

df_merged_1['AUM_Segment'] =  \
    np.where((df_merged_1['Revenue']>=0) & (df_merged_1['Revenue']<=2200), 'test1', 'test2')

answered Apr 23 '17 at 15:21

MaxU - stand with Ukraine

205,989
36
386
419

Python If Else Segmentation 'Invalid Type Comparison' Error in Dataframe

2 Answers2