0

在进行一些 OLS 回归时,我发现它statsmodels.api.add_constant()执行以下操作:

if _is_using_pandas(data, None) or _is_recarray(data):
    from statsmodels.tsa.tsatools import add_trend
    return add_trend(data, trend='c', prepend=prepend, has_constant=has_constant)

如果不是,它将被视为datandarray,因此您会丢失一些上下文信息(例如,作为回归变量名称的列名)。从 modin 导入 pandas 时,is_using_pandas()上述返回False.

可能statsmodels需要将其添加modin为受支持的选项,_is_using_pandas()但现在,我想做类似的事情:

if is_using_modin_pandas(x):
    from statsmodels.tsa.tsatools import add_trend
    X = add_trend(x, trend='c', prepend=True, has_constant='skip')
else:
    X = sm.add_constant(x)

一个人会怎么写is_using_modin_pandas()

4

0 回答 0