时间戳

首页 > 解决方案 > 方差膨胀因子输出统计模型

python - 方差膨胀因子输出统计模型

问题描述

我正在尝试使用 statsmodels 找到多重共线性，但我的代码输出并未显示方差膨胀因子，而是作为生成器对象的数据框：

from statsmodels.stats.outliers_influence import variance_inflation_factor
variables = df[['Mileage','Year','EngineV']]
vif = pd.DataFrame()
vif['VIF'] = (variance_inflation_factor(variables.values,i) for i in range(variables.shape[1]))
vif['features'] = variables.columns

结果输出

                                                 VIF  | features
  ---------------------------------------------------------------
0 | <generator object <genexpr> at 0x0000023A9F204... | Mileage
1 | <generator object <genexpr> at 0x0000023A9F204... |    Year
2 | <generator object <genexpr> at 0x0000023A9F204... | EngineV

而不是给出实际值。我确信这是一个简单的解决方法，但我对 Python 和编码非常陌生。谢谢

标签： pythonpandasstatisticsstatsmodels

解决方案

您应该将生成器转换为列表。您可以使用 list comprehansion 或 list() 函数。

vif['VIF'] = [variance_inflation_factor(variables.values,i) for i in range(variables.shape[1])]

或者

vif['VIF'] = list((variance_inflation_factor(variables.values,i) for i in range(variables.shape[1])))

推荐阅读