Statsmodels结合Patsy：如何保持回归参数的输入字符串顺序？

问题描述 #

将Patsy已完美集成于Statsmodels中，支持基于字符串编写R风格公式。我执行代码：

import statsmodels.formula.api as smf
res = smf.OLS.from_formula("Wealth ~ Age + Income + Happy", data=df).fit()
print(res.summary())

后，可查看回归结果摘要，但输出的参数顺序似乎无规律（示例：OLS Regression Results ============================================================================== Dep. Variable: Wealth R-squared: 0.309 Model: ...），请问如何让回归参数保持输入字符串指定的顺序？

解决方案 #

这个问题我之前做项目时也碰到过——Statsmodels默认会按变量名称的字母顺序排列参数，完全忽略你公式里写的顺序。下面分享几个实用的解决办法：

• 方法一：手动提取并重新排列参数
  拟合完成后，你可以先定义好想要的参数顺序，然后从结果中提取对应内容重新排列。比如：

  # 定义和公式一致的参数顺序，注意要包含截距项Intercept
  desired_order = ['Intercept', 'Age', 'Income', 'Happy']
  # 提取对应参数、p值等统计量
  ordered_params = res.params[desired_order]
  ordered_pvalues = res.pvalues[desired_order]
  # 如果要输出类似summary的格式，可以自己整理打印，比如：
  print("Ordered Regression Results:\n")
  for param, val, pval in zip(desired_order, ordered_params, ordered_pvalues):
      print(f"{param}: {val:.4f} (p-value: {pval:.4f})")
  

  这种方式简单直接，适合快速调整输出内容。

• 方法二：手动构建Patsy设计矩阵（推荐）
  本质上问题出在Patsy默认会对变量按字母排序，所以我们可以手动用Patsy生成设计矩阵，确保列顺序和公式一致，再传入OLS拟合：

  from patsy import dmatrices
  import statsmodels.api as sm
  
  # 生成y和X矩阵，return_type设为dataframe可以保留列名顺序
  y, X = dmatrices("Wealth ~ Age + Income + Happy", data=df, return_type='dataframe')
  # 用生成的X矩阵拟合OLS
  res = sm.OLS(y, X).fit()
  # 此时summary里的参数顺序就和公式完全一致了
  print(res.summary())
  

  这个方法能从根源上解决顺序问题，拟合后的所有结果（包括summary）都会严格遵循你公式里的变量顺序。

• 方法三：自定义Summary显示逻辑
  如果你想彻底修改Statsmodels默认的summary输出格式，可以继承它的Summary类重写参数表格的生成逻辑，但这个方法相对复杂，一般前两种就足够满足需求了。

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