1 回答

TA貢獻(xiàn)1801條經(jīng)驗(yàn) 獲得超8個(gè)贊
問題是您將分組數(shù)據(jù)幀傳遞到函數(shù)中pasty.dmatrices。由于分組數(shù)據(jù)幀是可迭代的,因此您可以在這樣的循環(huán)中執(zhí)行此操作,并將所有 X 數(shù)據(jù)幀(每組一個(gè))存儲(chǔ)到字典中:
import statsmodels.api as sm
import statsmodels.formula.api as smf
import numpy as np
import pandas as pd
import patsy
# Loading data
df = sm.datasets.get_rdataset("Guerry", "HistData").data
# Extracting Independent variables
formula = 'Suicides ~ Crime_parents + Infanticide'
data = df.groupby(['Region'])[['Suicides', 'Crime_parents', 'Infanticide', 'Region']]
X = {}
for name, group in data:
Y, X[name] = patsy.dmatrices(formula, group, return_type='dataframe')
print(X)
添加回答
舉報(bào)