2017-04-23 1 views
0

内の他のセグメンテーション「無効なタイプの比較」エラーの場合、私は、以下に示すサンプルデータフレーム(df_merged_1)を持っている:Pythonのデータフレーム

enter image description here

収益の列がのfloat64のDTYPEです。私は 'Revenue_Segment'という新しい列を作成したいと思います。

enter image description here以下

は私がセグメントに使用されるコードです::これは私が最終的な結果が見えるようにしたいものです

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 

ご協力いただきありがとうございます。

答えて

0

エレガントないが、これは私が今考えることができる唯一のソリューションです:

# 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 
+0

ブリリアント!これは完璧に働いた...ありがとうございます:) – PineNuts0

0

これを試してみてください:

df_merged_1['AUM_Segment'] = \ 
    np.where((df_merged_1['Revenue']>=0) & (df_merged_1['Revenue']<=2200), 'test1', 'test2') 
関連する問題