解决plotrix包pyramid.plot单条目时标签被强制改为'1'的问题

问题现象 #

使用plotrix包的pyramid.plot绘制人口金字塔时：

• 当左右侧（lx/rx）为多个条目时，自定义labels能正常显示
• 仅设置单一条目时，自定义标签会被强制替换为“1”
• 临时用矩阵传递标签会导致位置偏移，无法正常对齐

正常显示示例 #

library(plotrix)
pyramid.plot(c(10,5),
             c(7,5),
             labels=c("test","test1")
)

异常情况示例 #

library(plotrix)
pyramid.plot(10,
             5,
             labels="test"
)

问题根源 #

查看pyramid.plot源码，发现以下逻辑直接导致标签被强制替换：

ncats <- ifelse(!is.null(lxdim), dim(lx)[1], length(lx))
if (length(labels) == 1) 
    labels <- 1:ncats

当传入单条目数据时，ncats=1，此时如果labels长度为1，会被强制重置为1:1（即“1”），完全覆盖自定义标签。

修复方法（基于Base R，不更换工具） #

方法1：用trace函数修改函数局部逻辑 #

利用Base R的trace函数，拦截并替换掉导致标签被重置的代码块，仅保留默认标签的自动生成逻辑，手动传入的自定义单标签会被保留：

library(plotrix)

# 定义替换后的代码逻辑
fix_label_reset <- quote({
  ncats <- ifelse(!is.null(lxdim), dim(lx)[1], length(lx))
  # 仅当标签为默认的"1"时才生成序列，自定义标签直接保留
  if (length(labels) == 1 && labels == "1") 
    labels <- 1:ncats
})

# 将修改应用到pyramid.plot函数
trace(pyramid.plot, at = which(grepl("if (length(labels) == 1)", body(pyramid.plot))), 
      replace = fix_label_reset)

# 测试单条目自定义标签
pyramid.plot(10, 5, labels = "test")

方法2：构造长度匹配的标签向量 #

绕过源码的判断逻辑，传入长度为2的标签向量，其中第二个标签设为空字符串，再通过参数微调标签位置：

library(plotrix)
pyramid.plot(10,
             5,
             labels = c("test", ""),
             labelcex = 1, # 按需调整标签大小
             gap = 0.5 # 调整左右侧与标签的间距，修正位置偏移
)

这种方法不需要修改函数本身，通过构造符合函数逻辑的输入实现需求，标签位置可通过gap和labelcex参数灵活调整。

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