python - 尝试在 statsmodels python 中打印多重逻辑回归

Question

我正在尝试在 statsmodels 中打印一系列逻辑回归，但不确定如何将结果打印到控制台屏幕以外的其他地方。我创建了一个运行回归的函数，其中数据是数据集，其他变量是数据集中的一系列虚拟变量标签列表(我确信这不是运行这些的最有效方法，但它有效，并且当我有 5 个因变量，我想使用来自 var 2 和 var3 中的 15 的 9 个虚拟变量进行回归时，这是有意义的:

def regression(data, var1, var2, var3):
    for var in var 2:
        for var in var1:
            for var in var3:
                f = '%s ~ %s * %s' % (var1, var2, var3)
                y, x = patsy.dmatrices(f, data, return_type = 'dataframe')
                result = sm.Logit(y,x).fit()
                print(result.summary())
                print(np.exp(result.params))

无论如何 - 我正在尝试将输出打印到控制台以外的其他地方，以便我可以保存它并获得显着的结果。我敢打赌答案并不难，但我对 python 和 statsmodels 很陌生，并且在 statsmodels 文档或 stackoverflow 上没有看到任何足够明显的内容。谢谢

Answer 1

如果您想写入文件而不是标准输出，可以打开该文件并使用.write():

def regression(...):
    with open("outfile") as out:
        for var in var 2:
            for var in var1:
                for var in var3:
                    ...
                    out.write(str(result.summary()) + "\n")
                    out.write(str(np.exp(result.params)) + "\n)

正如@NeilG 指出的，您也可以使用print(result.summary(), out=out)。这几乎是 out.write(str(...) + "\n") 的快捷方式，如果你想把 result.summary() 和 np.exp(result.params) 在单独的文件中，使用 with open("summary", "w") 作为摘要，open("params", "w") 作为参数: 并使用 summary 和 params 而不是 out。