5

我有一个变量是一个因素：

 $ year           : Factor w/ 8 levels "2003","2004",..: 4 6 4 2 4 1 3 3 7 2 ...

我想创建 8 个虚拟变量，命名为“2003”、“2004”等，根据变量“year”的值取值 0 或 1。我能想到的最接近的是

dt1 <- cbind (dt1, model.matrix(~dt1$year - 1) )

但这会带来不幸的后果

  1. 虚拟变量命名为 dt1$year2003，而不仅仅是“2003”、“2004”等
  2. 似乎 NA 行被完全省略model.matrix（因此当变量中存在 NA 时，上述命令由于长度不同而失败year）。

当然，我可以用更多代码解决这些问题，但我希望我的代码尽可能简洁（在合理范围内），所以如果有人能提出更好的方法来制作虚拟变量，我将不胜感激。

4

3 回答 3

2

这是我能得到的最简洁的。该na.action选项负责处理NA值（我宁愿使用参数而不是全局选项设置，但我不知道如何）。列的命名是相当深刻的硬编码，看不到任何方法来覆盖它model.matrix...

options(na.action=na.pass)
dt1 <- data.frame(year=factor(c(NA,2003:2005)))
dt2 <- setNames(cbind(dt1,model.matrix(~year-1,data=dt1)),
              c("year",levels(dt1$year)))

如上所述，在某些情况下，列名不是合法的 R 变量名，您可能会遇到麻烦。

  year 2003 2004 2005
1 <NA>   NA   NA   NA
2 2003    1    0    0
3 2004    0    1    0
4 2005    0    0    1
于 2012-10-06T13:54:18.450 回答
2

您可以使用ifelse()which 不会省略na行（但我猜您可能不会将其视为“尽可能简洁”）：

dt1 <- data.frame(year=factor(rep(2003:2010, 10)))  # example data

dt1 <- within(dt1, yr2003<-ifelse(year=="2003", 1, 0))
dt1 <- within(dt1, yr2004<-ifelse(year=="2004", 1, 0))
dt1 <- within(dt1, yr2005<-ifelse(year=="2005", 1, 0))
# ...    

head(dt1)
#   year yr2003 yr2004 yr2005
# 1 2003      1      0      0
# 2 2004      0      1      0
# 3 2005      0      0      1
# 4 2006      0      0      0
# 5 2007      0      0      0
# 6 2008      0      0      0
于 2012-10-06T10:37:20.833 回答
1

library(caret) 提供了一个非常简单的函数 ( dummyVars) 来创建虚拟变量，尤其是当您有多个因子变量时。但是您必须确保目标变量是因素。例如，如果您Sales$year是数字，则必须将它们转换为因子：as.factor(Sales$year)

假设我们有原始数据集“销售”，如下所示：

    year    Sales       Region
1   2010    3695.543    North
2   2010    9873.037    West
3   2008    3579.458    West
4   2005    2788.857    North
5   2005    2952.183    North
6   2008    7255.337    West
7   2005    5237.081    West
8   2010    8987.096    North
9   2008    5545.343    North
10  2008    1809.446    West

现在我们可以同时创建两个虚拟变量：

>library(lattice)
>library(ggplot2)
>library(caret)
>Salesdummy <- dummyVars(~., data = Sales, levelsOnly = TRUE)
>Sdummy <- predict(Salesdummy, Sales)

结果将是：

   2005 2008 2010   Sales    RegionNorth    RegionWest
1   0    0    1   3695.543       1              0
2   0    0    1   9873.037       0              1
3   0    1    0   3579.458       0              1
4   1    0    0   2788.857       1              0
5   1    0    0   2952.183       1              0
6   0    1    0   7255.337       0              1
7   1    0    0   5237.081       0              1
8   0    0    1   8987.096       1              0
9   0    1    0   5545.343       1              0 
10  0    1    0   1809.446       0              1
于 2014-04-22T09:06:43.837 回答