将列名赋给值

Question

我和dplyr一起工作，我有一个类似的小消息：

df<- data_frame(one= c(1, NA, 1),
       two= c(NA,NA,1),
       three= c(1, 1,1)
       )
----------------------------
# A tibble: 3 x 3
      one   two  three
     <dbl> <dbl> <dbl>
 1     1    NA     1
 2    NA    NA     1
 3     1     1     1
----------------------------

----------------------------
# A tibble: 3 x 3
      one    two    three
     <dbl>  <dbl>   <dbl>
 1    one    NA     three
 2    NA     NA     three
 3    one    two    three
----------------------------

所以我可以对每个列使用ifelse函数和mutate：

df %>%
      one= ifelse(!is.na(one),'one', NA ),
      two= ifelse(!is.na(two),'two', NA ),
      three= ifelse(!is.na(three),'three', NA ),

但是在我真正的df中，我有很多列，那么这个wat是非常低效的。我需要一些更优雅的东西，使用mutate_at和列名，但这似乎很难。我试着用很多方法做到这一点，但每次我都会出错。有什么建议吗？

Answer 1

如果数据集中只有1和NAs，则将col(df)与数据集相乘，取消列表并根据索引，将其替换为数据集的名称并将其分配回原始数据

df[] <- names(df)[unlist(col(df)*df)]
df
# A tibble: 3 x 3
#    one   two three
#  <chr> <chr> <chr>
#1   one  <NA> three
#2  <NA>  <NA> three
#3   one   two three

或者使用TidyVerse，我们可以为每个列(MAP2_DFfromPurrr)执行此操作

library(tidyverse)
df %>%
     map2_df(names(.), ~replace(., !is.na(.), .y))
# A tibble: 3 x 3
#    one   two three
#  <chr> <chr> <chr>
#1   one  <NA> three
#2  <NA>  <NA> three
#3   one   two three