R语言自定义统计函数用cat输出时矩阵行列名丢失问题求助

问题根源 #

• cat() 函数仅用于控制台打印，返回值为NULL，将其与矩阵共同放入return()中拼接会直接破坏矩阵结构，导致维度名丢失、格式混乱
• 原代码中dimnames(a)引用了未定义变量a，本身会触发报错，也会影响矩阵维度名设置
• c()函数会将不同类型的对象强制转为一维向量，自然无法保留矩阵的二维结构

修正后代码 #

stat.descriptive <- function(var.cont, var.desc, titre) {
  # 按分类变量分组计算各描述统计量
  Moyenne <- round(by(var.cont, var.desc, mean, na.rm = TRUE), 2)
  Ecart_type <- round(by(var.cont, var.desc, sd, na.rm = TRUE), 2)
  Mediane <- round(by(var.cont, var.desc, median, na.rm = TRUE), 2)
  Minimum <- round(by(var.cont, var.desc, min, na.rm = TRUE), 2)
  Maximum <- round(by(var.cont, var.desc, max, na.rm = TRUE), 2)

  # 单独执行打印操作，不参与返回值构造
  cat(titre, "\n")
  cat("-----------------------------------------------", "\n")

  # 构造统计矩阵，自动继承行名（分类分组值）与列名（统计指标名）
  stat_matrix <- as.matrix(cbind(Moyenne, Ecart_type, Mediane, Minimum, Maximum))

  # 控制台打印完整矩阵结构
  print(stat_matrix)

  # 隐形返回矩阵对象，支持赋值调用
  return(invisible(stat_matrix))
}

核心修改说明 #

1. 打印逻辑与返回值逻辑完全分离：标题、分隔线仅用cat()做控制台输出，不放入return()中
2. 矩阵构造时无需额外设置dimnames，cbind()拼接的结果会自动保留分类变量的分组值作为行名，统计量变量名作为列名
3. 用invisible()返回矩阵，既不干扰控制台的打印顺序，又支持后续将统计结果赋值给变量做进一步处理

测试示例 #

可以直接用R内置的鸢尾花数据集验证效果：

stat.descriptive(
  var.cont = iris$Sepal.Length,
  var.desc = iris$Species,
  titre = "鸢尾花不同品种花萼长度描述统计结果"
)

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