要保留statsmodels中汇总结果中的变量名称,可以使用以下代码示例:
import statsmodels.api as sm
# 构建线性回归模型
X = sm.add_constant(X) # 添加常数列
model = sm.OLS(y, X)
results = model.fit()
# 提取变量名称
variable_names = results.params.index.tolist()
# 输出变量名称
print(variable_names)
上述代码假设你已经定义了自变量X和因变量y,并且通过sm.add_constant()
方法添加了常数列。然后,使用sm.OLS()
方法构建了一个线性回归模型,并通过fit()
方法拟合了模型。
最后,通过results.params.index.tolist()
方法提取了模型的参数名称,并将其存储在变量variable_names
中。你可以根据需要进一步处理或输出这些变量名称。
请注意,上述代码示例中的results
是OLS
对象的一个实例,你可以根据自己的模型类型进行相应的更改。