在 for 循環中附加帶有新鍵和值的字典

Question

我目前正在嘗試循環通過 dataframe 進行線性回歸。 每次迭代后，我想保存當前列名和生成的 Adj。 字典中的 R 平方值：

import statsmodels.api as sm

for columns in [x for x in var_sel.columns if x != 'price']:
    
    fit_d = {}
    
    Y = var_sel['price']

    X = var_sel[columns]

    X = sm.add_constant(X)

    model = sm.OLS(Y,X, missing = 'drop').fit()
    
    fit_d[columns] = model.rsquared

但是當我打印字典時，它只輸出{'room_type_Shared room': 0.0003039648861731248}這是 dataframe 的最后一列。 我如何 append 字典與每個迭代結果？

Answer 1

您必須在 for 循環之外聲明字典。 在上面的代碼中，每次循環運行時都會重新聲明字典。

import statsmodels.api as sm

fit_d = {}

for columns in [x for x in var_sel.columns if x != 'price']:
    
    Y = var_sel['price']

    X = var_sel[columns]

    X = sm.add_constant(X)

    model = sm.OLS(Y,X, missing = 'drop').fit()
    
    fit_d[columns] = model.rsquared