如何在 R 中有条件地粘贴因子标签

Question

希望有人能帮助我解决这个问题，我自己找到了一个糟糕的解决方案：我想聚合（或粘贴）四列的标签（A 到 D) 变成第五个 (dream) 但有条件地，只有当它的数值是 2 时。

这是我的数据库 df：

id= c(1:12)
A = c(2,NA,NA,2,NA,1,1,1,1,1,NA,2)
B = c(2,1,1,1,2,NA,1,1,1,1,2,1)
C = c(2,1,1,1,2,NA,1,1,1,1,NA,1)
D = c(2,1,1,1,1,1,2,1,1,NA,2,1)
df = data.frame(id,A,B,C,D) ; df
df$A=factor(df$A, labels=c("no", "i saw"))
df$B=factor(df$B, labels=c("no", "someone"))
df$C=factor(df$C, labels=c("no", "sitting"))
df$D=factor(df$D, labels=c("no", "on a cloud")) ; df

下面是我找到的解决方案，但不太令人满意...

df$dream = ifelse(as.numeric(df$A)!=2, NA, as.character(df$A)) ; df
df$dream = ifelse(as.numeric(df$B)!=2, df$dream, paste(df$dream, as.character(df$B))) ; df
df$dream = ifelse(as.numeric(df$C)!=2, df$dream, paste(df$dream, as.character(df$C))) ; df
df$dream = ifelse(as.numeric(df$D)!=2, df$dream, paste(df$dream, as.character(df$D))) ; df

我确信有一种直接的方法可以做到这一点，此外我的代码似乎甚至不能以这种方式工作。有人可以帮助我吗？谢谢

Answer 1

我觉得如果你把你data.frame转换成data.table会更容易。

对于column B你可以使用

dt <- as.data.table(df)
dt[,dream:=ifelse(B==2,"someone",ifelse(B==1,"no",NA))]

然后对其余部分进行相同的复制 3 columns。希望对您有所帮助

更新

或者你可以试试这个

dt$dream.A <- mapvalues(dt$A,c(1,2),c("no", "i saw"))

Answer 2

此解决方案可行，但您必须声明要从因子粘贴的值向量。

# init empty result vector
dream <- character(nrow(df))
# values from each column (A-D) you want to paste
values <- c("i saw","someone","sitting", "on a cloud")
# iterate over each row
for(i in seq_len(nrow(df))){
  #paste values from each row
  dream[i] <- paste(values[which(as.numeric(df[i,-1]) == 2)], collapse=" ")  
}

如何在 R 中有条件地粘贴因子标签

How to paste factor labels conditionally in R

conditional

r

concatenation

paste