Python如果其他分段'无效类型比较' Dataframe中的错误

时间:2017-04-23 14:14:53

标签: if-statement dataframe comparison segment

我有一个示例数据框(df_merged_1),如下所示:

enter image description here

Revenue列是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

非常感谢任何帮助!

2 个答案:

答案 0 :(得分: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 

答案 1 :(得分:0)

试试这个:

table {
 border-collapse:collapse;
 width: 100%; 
}

td {
 background: #eee;  
 border-bottom: 1px solid #000;
 padding:5px 10px
}

tr:last-child td{
 border-bottom:none
}

<table>
 <tr>
 <td>a</td>
 <td>b</td>
 <td>c</td>
 <td>d</td>
 <td>e</td>
</tr>
<tr>
 <td>a</td>
 <td>b</td>
 <td>c</td>
 <td>d</td>
 <td>e</td>
</tr>
<tr>
 <td>a</td>
 <td>b</td>
 <td>c</td>
 <td>d</td>
 <td>e</td>
</tr>
</table>