使用Purrr :: map2循环两个列名向量,以便将多列重新编码为新变量

时间:2017-10-26 18:45:48

标签: r tidyverse purrr

library(tidyverse)    

使用下面的示例代码,我想使用" mutate"或" mutate_at"根据另一列的值将多列重新编码为新列。基本上,我想重新编码结束的变量是" s" (q25s,q26s等......),基于相应的非"#34;变量。因此,例如,如果q25 = 1,则将重新编码q25s,使得1 = 0,2 = 0,3 = 0,4 = 1,5 = 1,并且88 =丢失,并且新名称将是q25_new。如果q25不等于1,则不应该重新编码,q25_new应该只是NA。

但是,为了实现这一点,我尝试使用tidyverse创建列名的命名向量,然后使用" mutate"," recode",和& #34; if_else"和purr :: map2一起。

我认为下面的代码应该是可能的吗?我无法让它工作......而且我觉得我需要使用" paste0"在某个地方命名所有以" _new"开头的新列变量名称。

cols1<-Df %>%select(q25:q29)
cols2<-Df %>% select(q25s:q29s)

Df<- Df %>% map2(Df[cols1],Df[cols2],
  ~if_else(.x==1, mutate_at(vars (.y),funs(recode(.,`1`=0,`2`=0,`3`=0,`4`=1,`5`=1),"NA"))))

以下是示例代码。

Here is the sample code:
q25<-c(2,1,88,2,1)
q26<-c(2,88,88,88,2)
q27<-c(2,2,1,1,1)
q28<-c(88,1,1,2,2)
q29<-c(1,1,1,2,2)
q25s<-c(3,5,88,4,1)
q26s<-c(4,4,5,5,1)
q27s<-c(3,3,4,1,4)
q28s<-c(4,5,88,1,3)
q29s<-c(88,88,3,4,4)
Df<-data.frame(q25,q26,q27,q28,q29,q25s,q26s,q27s,q28s,q29s)

1 个答案:

答案 0 :(得分:1)

这会有用吗?

map2(Df[1:5],Df[6:10], ~ if_else(.x==1, recode(.y,`1`=0,`2`=0,`3`=0,`4`=1,`5`=1,`88` = NA_real_),NA_real_)) %>%
  as.data.frame %>%
  rename_all(paste0,"_new") %>%
  cbind(Df,.)

#   q25 q26 q27 q28 q29 q25s q26s q27s q28s q29s q25_new q26_new q27_new q28_new q29_new
# 1   2   2   2  88   1    3    4    3    4   88       3       4       3       4      NA
# 2   1  88   2   1   1    5    4    3    5   88       1       4       3       1      NA
# 3  88  88   1   1   1   88    5    4   88    3      88       5       1      NA       0
# 4   2  88   1   2   2    4    5    1    1    4       4       5       0       1       4
# 5   1   2   1   2   2    1    1    4    3    4       0       1       1       3       4

好吧最后我无法抗拒挑战,所以这里几乎是100%整齐的方式(同样的输出):

library(tidyr)
Df %>%
  mutate(n=row_number()) %>%
  gather(key,value,-n) %>%
  mutate(key2 = ifelse(grepl("s",key),"s","x"),
         key=sub("s","",key)) %>%
  spread(key2,value) %>%
  mutate(`_new` = if_else(x==1, recode(s,`1`=0,`2`=0,`3`=0,`4`=1,`5`=1,`88` = NA_real_),Inf)) %>%
  gather(key3,value,s,x,`_new`) %>%
  unite(key,key,key3,sep="") %>%
  spread(key,value) %>%
  rename_all(~gsub("x","",.x)) %>%
  select(order(nchar(names(.))),-n)