如何在R基础绘图中按lasso系数重排并翻转回归估计图

解决方案步骤 #

要实现类似ggplot2::coord_flip()的垂直翻转效果，同时让回归估计值最高的类别位于图的顶部，只需对数据排序后互换坐标轴，并自定义轴位置即可，具体操作如下：

1. 数据预处理：按估计值降序排序
   首先将数据框按lasso.coef降序排列，确保最大的估计值对应的类别排在最前面：

   # 按lasso.coef降序排序，最大估计值的类别优先
   sorted_df <- lasso.model.bin.df.nozero[order(-lasso.model.bin.df.nozero$lasso.coef), ]
   # 为每个类别分配y轴位置，让最大估计值的类别位于图顶部
   sorted_df$y_pos <- nrow(sorted_df):1
   

2. 基础绘图实现翻转与排序
   调整绘图参数，互换原x/y轴的角色，并对应修改辅助线和轴标签：

   # 设置绘图边距（左侧预留足够空间显示类别标签）
   par(mar=c(6, 10, 2, 2), mgp=c(3, 1, 0))
   
   # 绘制散点图：x为回归估计值，y为自定义的类别位置
   plot(sorted_df$lasso.coef, sorted_df$y_pos, 
        col=sorted_df$color, pch=16, las=1, 
        xlab="聚合二项式回归估计值\n(经Lasso正则化保留的动词)", 
        ylab="", yaxt="n", main="", cex.main=1)
   
   # 添加截距垂直线（原水平截距线因轴翻转变为垂直）
   abline(v=lasso.model.bin.intercept, lty="longdash", col="black")
   # 添加水平虚线分隔每个类别（原垂直分隔线因轴翻转变为水平）
   abline(h=sorted_df$y_pos, lty="dotted", col="grey")
   
   # 添加y轴类别标签，水平显示避免重叠
   axis(side=2, at=sorted_df$y_pos, labels=sorted_df$verb, las=1, cex.axis=0.75)
   

关键说明 #

• 轴翻转原理：通过将原y轴的回归估计值设为新的x轴，原x轴的类别位置设为新的y轴，实现垂直翻转效果。
• 顶部排序逻辑：用nrow(sorted_df):1生成倒序的y轴位置，让排序后第一个（最大估计值）的类别对应y轴的最高刻度，从而显示在图的顶部。
• 标签优化：设置las=1让y轴标签水平显示，同时调大左侧边距，避免标签被截断。

内容的提问来源于stack exchange，提问作者nick.yili9393