在进行一些 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()?