的Python如果否則分割「無效類型的比較」在數據幀

Question

錯誤我有以下所示的樣品數據幀（df_merged_1）：

的收益列是float64 D型。我想創建一個名爲'Revenue_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' 

但代碼是不工作...我得到以下錯誤：

TypeError: invalid type comparison 

任何幫助，非常感謝！

Answer 1

不優雅，但是這是我現在能想到的唯一的解決辦法：

# 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 

Answer 2

試試這個：

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