루프 (파이썬)에서 함수의 인수를 변경하는 방법은 무엇입니까?

Question

Python에서 Y, X1, X2, X3으로 표시된 열이있는 판다 데이터 프레임이 있다고 가정합니다. X에서 X1, X2, X3으로 OLS 회귀 분석을 실행합니다. 루프를 사용하여 회귀에서 계수를 목록에 저장하고 싶습니다. 루프에서 함수의 인수를 변경하려면 어떻게합니까? 다음 코드는 작동하지 않습니다.

import pandas as pd 
import statsmodels.api as sm 

ols = sm.OLS(df[['Y']],(df[['X_1','X_2','X_3']])) 
olsfit = ols.fit(cov_type='HC0') 

coef = [] 

for i in range(4): 
    coef.append(olsfit.params.X_i) 

Answer 1

나는이 코드가 작동하게하기에 충분 있는지 확실하지 않습니다,하지만 당신은 방금 말한 getattr

# Also using comprehension notation 
coef = [ 
    getattr(olsfit.params, 'X_' + str(i)) 
    for i in range(4) 
] 

를 사용하여 동적으로 속성에 액세스해야하는 경우, 당신이 따라 제안

Answer 2

대신 회귀 이름을 구성하는 정수 반복의 아코의 대답은 단지 북동 인합니다 (statsmodels에 RegressionResultsWrapperparams 속성을 실제 이름을 사용 pd.Series, 회귀 변수 이름에 색인 됨) - 적절하게 필터링하십시오.

olsfit.params[olsfit.params.index.str.contains('^X_')]