使用NA的条件语句填充数据框中列的值-R

时间:2018-07-05 10:12:13

标签: r if-statement conditional na

我有一个数据框,如下所示:

Date        FLOW     Modelled   Infilled
01-01-1992  1.856    1.900      NA
02-01-1992  1.523    1.500      NA
03-01-1992  NA       2.400      NA    
04-01-1992  3.679    3.800      NA

我想用FLOW值填充“已填充”列。在时间序列的“流量”列中有“ NA”值的地方,我想用“建模”列中的值替换这些NA。

答案应如下所示:

Date        FLOW     Modelled   Infilled
01-01-1992  1.856    1.900      1.856
02-01-1992  1.523    1.500      1.523
03-01-1992  NA       2.400      2.400    
04-01-1992  3.679    3.800      3.679

我在excel中有以下解决方案:

Infilled column   =IF((FLOW="NA"),Modelled,FLOW)

我尚未找到在线解决方案来帮助我在R中进行编程。时间序列非常长,我需要执行多个文件,因此循环可能是最合适的解决方案。我对R比较陌生,我无法弄清楚。帮助非常感谢!

2 个答案:

答案 0 :(得分:1)

您正在寻找coalesce

library(tidyverse)
dat%>%
   mutate(Infilled=coalesce(FLOW,Modelled))
        Date  FLOW Modelled Infilled
1 01-01-1992 1.856      1.9    1.856
2 02-01-1992 1.523      1.5    1.523
3 03-01-1992    NA      2.4    2.400
4 04-01-1992 3.679      3.8    3.679

在base R中,您可以执行以下操作:

transform(dat,Infilled=ifelse(is.na(FLOW),Modelled,FLOW))
        Date  FLOW Modelled Infilled
1 01-01-1992 1.856      1.9    1.856
2 02-01-1992 1.523      1.5    1.523
3 03-01-1992    NA      2.4    2.400
4 04-01-1992 3.679      3.8    3.679

答案 1 :(得分:1)

我们可以使用base R

dat$Infilled <- dat$FLOW
i1 <- is.na(dat$FLOW)
dat$Infilled[i1] <- dat$Modelled[i1]

或与data.table

library(data.table)
setDT(dat)[, Infilled := FLOW][is.na(FLOW), Infilled := Modelled][]