含NA起始列的新列计算问题（tidyverse/dplyr环境）

问题核心是R中NA与任何值运算都会返回NA，要实现NA位置结果为1，只需将col2中的NA替换为0后再与col1相加（因为1+0=1），以下是两种基于tidyverse/dplyr的实现方式：

方法1：使用coalesce函数（推荐，更简洁） #

coalesce会返回第一个非NA值，直接将NA替换为0后计算：

library(tidyverse)

# 构造数据框
df <- data.frame(col1 = 1:4, col2 = c(NA,1,2,3), col3 = 5:8)

# 生成新列
df %>% 
  mutate(new = col1 + coalesce(col2, 0))

运行结果：

col1 col2 col3 new
1    1   NA    5   1
2    2    1    6   3
3    3    2    7   5
4    4    3    8   7

方法2：使用replace_na函数 #

先显式替换列中的NA为0，再执行加法运算：

df %>% 
  mutate(
    new = col1 + replace_na(col2, 0)
  )

适配列索引的写法 #

如果你需要保留用列索引（.[[1]]/.[[2]]）的方式，同样可以用coalesce：

df %>% 
  mutate(new = .[[1]] + coalesce(.[[2]], 0))

内容的提问来源于stack exchange，提问作者frank