如何使用`[`函数替换向量中的NA值？

为什么管道代码和直接赋值效果不同？ #

你的管道代码x <- x %>% [(is.na(.)) %>% <-("e")逻辑错误，执行过程是：

1. 第一步x %>% [(is.na(.))：调用[函数提取x中的NA元素，得到单元素向量NA
2. 第二步将这个NA向量传给<-函数，相当于执行(x[is.na(x)]) <- "e"，这个表达式的返回值是赋值内容"e"
3. 最后把这个返回值赋值给x，所以x变成仅含"e"的单元素向量

而x[is.na(x)] <- "e"是R的子集赋值语法，它不是普通函数调用：R会直接定位x中NA对应的位置，把"e"写入这些位置，最终x保留原长度，仅指定位置被修改。

[作为函数的工作原理 #

[是R内置的普通函数，我们常用的x[i]、df[, col]都是该函数的语法糖，等价写法如下：

• 向量索引：[(x, i)
• 数据框索引：[(df, row_idx, col_idx)

它的核心行为是：接收原对象和索引参数，返回原对象的子集，不会修改原对象本身。

举几个直观例子：

x <- c("a", "b", "c", NA)
# 等价于 `[`(x, 2)
x[2]  # 返回 "b"
# 等价于 `[`(x, is.na(x))
x[is.na(x)]  # 返回 NA

正确的管道写法 #

如果想用管道实现NA替换，有两种常用方式：

1. 使用replace函数（推荐） #

replace函数专门用于替换向量指定元素，适配管道逻辑：

x <- c("a", "b", "c", "d", NA, "f", "g")
x <- x %>% replace(is.na(.), "e")
# 结果：c("a", "b", "c", "d", "e", "f", "g")

2. 使用复合赋值管道%<>%（需magrittr包） #

%<>%是magrittr包提供的复合赋值管道，会把管道处理后的结果重新赋值给原变量：

library(magrittr)
x <- c("a", "b", "c", "d", NA, "f", "g")
x %<>% { .[is.na(.)] <- "e"; . }

这里用大括号{}包裹代码块，先修改子集，再返回修改后的完整向量。

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