R 来自 Cols 的第一个非 NA 值

Question

df <- data.frame(ID=c(1,2,3,4,5,6),
                CO=c(-6,4,2,3,0,2),
                CATFOX=c(1,NA,NA,3,0,NA),
                DOGFOX=c(NA,NA,5,1,2,NA),
                RABFOX=c(NA,3,NA,5,3,NA),
                D=c(0,4,5,6,1,2),
                WANT=c(1,3,5,3,0,NA))

我有一个数据框，我希望让列 WANT 取 'CATFOX' 'DOGFOX' 'RABFOX' 的第一个不是 NA 的值。有data.table解决方案吗？我试过了，但没有产生预期的结果：

df$WANT=do.call(coalesce, data[grepl('FOX',names(data))])

Answer 1

您可以试试这个base R解决方案：

#Data
data=data.frame(ID=c(1,2,3,4,5),
                CO=c(-6,4,2,3,0),
                CATFOX=c(1,NA,NA,3,0),
                DOGFOX=c(NA,NA,5,1,2),
                RABFOX=c(NA,3,NA,5,3),
                D=c(0,4,5,6,1),
                WANT=c(1,3,5,3,0))
#Process
index <- which(names(data) %in% c('CATFOX','DOGFOX','RABFOX'))
data$WANT2 <- apply(data[,index],1,function(x) x[min(which(!is.na(x)))])

输出：

  ID CO CATFOX DOGFOX RABFOX D WANT WANT2
1  1 -6      1     NA     NA 0    1     1
2  2  4     NA     NA      3 4    3     3
3  3  2     NA      5     NA 5    5     5
4  4  3      3      1      5 6    3     3
5  5  0      0      2      3 1    0     0

Answer 2

您的示例中有 coalesce，这是 dplyr 的构造。试试 fcoalesce:

library(data.table)

setDT(df)[, WANT2 := fcoalesce(CATFOX, DOGFOX, RABFOX)]

输出：

   ID CO CATFOX DOGFOX RABFOX D WANT WANT2
1:  1 -6      1     NA     NA 0    1     1
2:  2  4     NA     NA      3 4    3     3
3:  3  2     NA      5     NA 5    5     5
4:  4  3      3      1      5 6    3     3
5:  5  0      0      2      3 1    0     0
6:  6  2     NA     NA     NA 2   NA    NA

Answer 3

我们可以在 base R

中使用矢量化选项

i1 <- endsWith(names(df), 'FOX')
df$WANT2 <-  df[i1][cbind(seq_len(nrow(df)), max.col(!is.na(df[i1]), 'first'))]
df$WAN2
#[1]  1  3  5  3  0 NA

R 来自 Cols 的第一个非 NA 值

R First Non-NA Value From Cols

r

coalesce

data.table