基于值的数据重新编码一个新的分类标签列

Question

我有一个每天记录商品价格的数据集。基于价格数据，我想编码一个新的“action”变量。这个变量应该用-1、0或1标记每行价格数据。这些数字对应于“卖出、不做任何事、买入”。我可以在R或Python中完成这项工作。最有效的方法是什么

样本输入：

8/26/2017, 5
8/27/2017, 5
8/28/17, 1
8/29/17, 3
8/30/17, 5 

样本输出：

8/26/2017, -1
8/27/2017, -1
8/28/17, 1
8/29/17, 0
8/30/17, -1 

Answer 1

我们可以使用match在R中对其进行索引

df1$price <- c(-1, 1, 0)[match(df1$price, c(5, 1, 3))]
df1$price
#[1] -1 -1  1  0 -1

数据

df1 <- structure(list(date = c("8/26/2017", "8/27/2017", "8/28/17", 
"8/29/17", "8/30/17"), price = c(5, 5, 1, 3, 5)), .Names = c("date", 
"price"), class = "data.frame", row.names = c(NA, -5L))