如何在含缺失值时用R包flextable按值为单元格着色

当使用flextable对包含NA的数据集应用条件单元格着色时，会触发报错：

Error in data_colors(source_dataset, color) : colors can not contain missing values

原因是自定义颜色函数my_color_fun在遇到NA值时，dplyr::if_else会返回NA，而flextable的color函数不允许颜色向量中存在缺失值。

解决方法：显式处理NA值 #

只需修改自定义颜色函数，明确指定NA值对应的颜色即可，以下提供两种实现方式：

方式1：利用if_else的missing参数

dplyr::if_else自带missing参数，可直接指定输入为NA时的返回值：

library(flextable)
library(dplyr)

my_color_fun <- function(x) {
  # 用missing参数指定NA对应的颜色
  if_else(x > 4 & x < 5, "red", "black", missing = "black")
}

# 引入NA的数据集
df <- iris
df$Sepal.Length[5] <- NA

# 正常生成带条件着色的表格
flextable(df) %>%
  color(j = c("Sepal.Length", "Sepal.Width", "Petal.Length", "Petal.Width"), color = my_color_fun)

方式2：用case_when清晰定义多条件

如果后续需要扩展更多着色规则，case_when的可读性更强，可单独为NA值设置规则：

library(flextable)
library(dplyr)

my_color_fun <- function(x) {
  case_when(
    x > 4 & x < 5 ~ "red",
    is.na(x) ~ "gray", # 可自定义NA单元格的颜色，比如灰色
    TRUE ~ "black"
  )
}

# 引入NA的数据集
df <- iris
df$Sepal.Length[5] <- NA

# 正常生成带条件着色的表格
flextable(df) %>%
  color(j = c("Sepal.Length", "Sepal.Width", "Petal.Length", "Petal.Width"), color = my_color_fun)

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