case_when可以报告它选择的变量的名称以及它们的值吗？

Question

我有一个数据集，对于每个唯一的DOI，一个特定变量的值分散在三个变量(列)的任意组合中。之所以出现这种情况，是因为我搜索了三个不同的在线搜索引擎来查找DOI信息，并将结果合并到一个文件中。一个示例数据集如下：

df1 <- data.frame(
  'DOI' = c("A","B","C","D","E", "G", "H", "I")
  , 'V2' = c(1,NA,NA,4,5,NA,7, NA)
  , 'V3' = c(NA,2,NA,4,5,6,NA, NA)
  , 'V4' = c(NA,NA,3,4,NA,6,7, NA)
)

df1

  DOI V2 V3 V4
1   A  1 NA NA
2   B NA  2 NA
3   C NA NA  3
4   D  4  4  4
5   E  5  5 NA
6   G NA  6  6
7   H  7 NA  7
8   I NA NA NA

以下代码允许我选择包含在其中的三列(V2、V3或V4)中的数据，并将其保存在单个新变量V5中。

library(dplyr)
df2 <- df1 %>% mutate(V5 = case_when((!is.na(V2) & is.na(V3) & is.na(V4)) ~ V2,
                            (is.na(V2) & !is.na(V3) & is.na(V4) ) ~ V3,
                            (is.na(V2) & is.na(V3) & !is.na(V4) ) ~ V4,
                            (!is.na(V2) & !is.na(V3) & !is.na(V4) ) ~ V2,
                            (!is.na(V2) & !is.na(V3) & is.na(V4) ) ~ V3,
                            (is.na(V2) & !is.na(V3) & !is.na(V4) ) ~ V3,
                            (!is.na(V2) & is.na(V3) & !is.na(V4) ) ~ V2,
                            (is.na(V2) & is.na(V3) & is.na(V4) ) ~ V2,
                            TRUE ~ NA_real_))
df2

  DOI V2 V3 V4 V5
1   A  1 NA NA  1
2   B NA  2 NA  2
3   C NA NA  3  3
4   D  4  4  4  4
5   E  5  5 NA  5
6   G NA  6  6  6
7   H  7 NA  7  7
8   I NA NA NA NA

我还想记录(作为V6)，哪些变量(V2、V3或V4)是复制到V5中的数据的来源。我尝试了以下操作，但它似乎只是复制了进入V5的变量的值，而不是变量本身的名称。

df3 <- df2 %>% mutate(V6 = case_when((!is.na(V2) & is.na(V3) & is.na(V4)) ~ "V2",
                                     (is.na(V2) & !is.na(V3) & is.na(V4) ) ~ "V3",
                                     (is.na(V2) & is.na(V3) & !is.na(V4) ) ~ "V4",
                                     (!is.na(V2) & !is.na(V3) & !is.na(V4) ) ~ "V2",
                                     (!is.na(V2) & !is.na(V3) & is.na(V4) ) ~ "V3",
                                     (is.na(V2) & !is.na(V3) & !is.na(V4) ) ~ "V3",
                                     (!is.na(V2) & is.na(V3) & !is.na(V4) ) ~ "V2",
                                     (is.na(V2) & is.na(V3) & is.na(V4) ) ~ "V2",
                                     TRUE ~ NA_real_))
df3

  DOI V2 V3 V4 V5 V6
1   A  1 NA NA  1  1
2   B NA  2 NA  2  2
3   C NA NA  3  3  3
4   D  4  4  4  4  4
5   E  5  5 NA  5  5
6   G NA  6  6  6  6
7   H  7 NA  7  7  7
8   I NA NA NA NA NA

是否也可以记录从其中选择数据的变量的名称？这将允许跟踪我用来定位匹配的DOI信息的搜索引擎中的哪个。

Answer 1

下面是一个基本的R选项，它使用max.col来选择每行中的第一个非NA值和相应的列名。

inds <- max.col(!is.na(df1[-1]), 'first')
df1$V5 <- df1[-1][cbind(1:nrow(df1), inds)]
df1$V6 <- names(df1[-1])[inds]
#To replace column name with all NA value to NA
df1$V6[rowSums(!is.na(df1[-1])) == 0] <- NA

如果您对dplyr解决方案感兴趣，可以在c_across中使用rowwise：

library(dplyr)

df1 %>%
  rowwise() %>%
  mutate(V5 = na.omit(c_across(V2:V4))[1], 
         V6 = names(.)[-1][which(!is.na(c_across(V2:V4)))[1]])

#   DOI      V2    V3    V4    V5 V6   
#  <chr> <dbl> <dbl> <dbl> <dbl> <chr>
#1 A         1    NA    NA     1 V2   
#2 B        NA     2    NA     2 V3   
#3 C        NA    NA     3     3 V4   
#4 D         4     4     4     4 V2   
#5 E         5     5    NA     5 V2   
#6 G        NA     6     6     6 V3   
#7 H         7    NA     7     7 V2   
#8 I        NA    NA    NA    NA NA   

我们使用-1是因为我们想忽略计算中的第一列DOI。

Answer 2

使用R基：

> df1
  DOI V2 V3 V4
1   A  1 NA NA
2   B NA  2 NA
3   C NA NA  3
4   D  4  4  4
5   E  5  5 NA
6   G NA  6  6
7   H  7 NA  7
8   I NA NA NA
> df1$V5 <- apply(df1[2:4], 1, max, na.rm = T)
> df1$V5[is.infinite(df1$V5)] <- NA
> df1$V6 <- apply(df1[2:4],1,which.max)
> df1$V6 <- as.integer(df1$V6)
> df1$V6 <- names(df1)[2:4][df1$V6]
> df1
  DOI V2 V3 V4 V5   V6
1   A  1 NA NA  1   V2
2   B NA  2 NA  2   V3
3   C NA NA  3  3   V4
4   D  4  4  4  4   V2
5   E  5  5 NA  5   V2
6   G NA  6  6  6   V3
7   H  7 NA  7  7   V2
8   I NA NA NA NA <NA>
>