Python为特定值添加id

时间:2018-03-30 05:03:01

标签: python dataframe

如何在数据框中为id添加列?介于0到100之间的值的id应为1,否则为2。

  time                    values
 2018-03-19 14:31:17.200  1095
 2018-03-19 14:31:17.300  2296  
 2018-03-19 14:31:17.400  2147   
 2018-03-19 14:31:17.500   309   
 2018-03-19 14:31:17.600   244 
 2018-03-19 14:31:17.700   263  
 2018-03-19 14:31:17.800   548  

1 个答案:

答案 0 :(得分:1)

我认为numpy.where创建的条件需要between(默认inclusive=True):

df['id'] = np.where(df['values'].between(0,100), 1,2)
print (df)
                      time  values  id
1  2018-03-19 14:31:17.200    1095   2
2  2018-03-19 14:31:17.300    2296   2
3  2018-03-19 14:31:17.400    2147   2
4  2018-03-19 14:31:17.500     309   2
5  2018-03-19 14:31:17.600     244   2
6  2018-03-19 14:31:17.700     263   2
7  2018-03-19 14:31:17.800     548   2