使用pandas和statsmodels时拉取变量名称

import pandas as pd

# 创建一个DataFrame对象
data = {'A': [1, 2, 3], 'B': [4, 5, 6], 'C': [7, 8, 9]}
df = pd.DataFrame(data)

# 获取DataFrame对象中的所有变量名称
variable_names = list(df.columns)
print(variable_names)

['A', 'B', 'C']

import pandas as pd
import statsmodels.api as sm

# 创建一个DataFrame对象
data = {'A': [1, 2, 3], 'B': [4, 5, 6], 'C': [7, 8, 9]}
df = pd.DataFrame(data)

# 使用Formula对象创建模型
formula = 'A ~ B + C'
model = sm.formula.ols(formula, data=df).fit()

# 获取模型中的所有变量名称
variable_names = model.exog_names
print(variable_names)

['Intercept', 'B', 'C']