r - 更改 R 中虚拟变量的名称？

Question

我最近问了一个类似的问题here .但是，我有一个新问题，我无法应用其他答案。我正在尝试根据名称向量更改列元素的名称。

我遇到的问题是，如果我有一个包含如下因子变量的数据框:

dfOG <- data.frame(
  x1 = c(1,4,2),
  x4 = as.factor(c('yes', 'no', 'maybe')),
  x5 = as.factor(c(rep('yes', 2), rep('no', 1))),
  x41 = as.factor(c(rep('hot', 1), rep('cold', 2)))
)

然后我将应用三个过程之一。每个过程(不在我的控制范围内)都会将这些因素分解为虚拟变量。但不幸的是，根据所使用的过程，它会以不同的方式重命名假人。例如，下表显示了如何拆分 x4，具体取决于所使用的过程:

x41, x42, x43 = x4 # process 1
x4.yes, x4.no, x4.maybe = x4 # process 2
x4_yes, x4_no, x4_maybe = x4 # process 3

因此，进程 1 只是在末尾添加一个数字。流程 2 添加 . 后跟因子级别。过程 3 添加一个 _，后跟因子级别。

只关注过程 1，在应用该过程后，我最终得到一个数据框，其中有一列如下所示:

dfChange <- data.frame(
  name = c('x41', 'x43', NA, NA,
           'x411', NA, 'x52',
           'x42', 'x412', NA)
)

我正在尝试将 dfChange$name 中的元素名称更改回它们在 dfOG 中的原始名称。

期望的输出

我想要的输出看起来像这样:

dfDesired <- data.frame(
  name = c('x4', 'x4', NA, NA, 
           'x41', NA, 'x5',
           'x4', 'x41', NA)
)

尝试的解决方案

我实际上还没有解决方案，但这是我正在尝试的:

# find out which columns in my original data are factors
factorColNam <- names(which(!(sapply(dfOG[colnames(dfOG)], is.numeric))))
factorCols <- which((colnames(dfOG) %in% factorColNam))


# create a list of the variables split into their factors
dfnew <- list()
for (i in 1:length(factorCols)) {
  facLevels <- unique(dfOG[ ,factorCols[i]])
  dfnew[[i]] <- paste0(factorColNam[i],  as.numeric(facLevels))
}

# rename each element it's original name
names(dfnew) <- factorColNam

我的想法是然后遍历列表元素并比较 dfChange$name 中的名称并将它们重命名为它们所在的任何列表元素。但我确定必须有一个更简单更好的方法？

编辑

我忘了说的是名称可以是任何东西，它们不限于 x1, x2,... 等。所以，例如，我可以有一个名为 x10 具有 10 个因子水平，或者名为 banana 的变量具有 25 个因子水平。任何名称都可以。

Answer 1

从您的代码继续:

dfChange$var <- NA
for (i in 1:length(dfnew)) {
  nn <- names(dfnew)
  dfChange$var[which(dfChange$name %in% dfnew[[i]])] <- nn[[i]]
}