팬더 데이터 프레임의 열을 반복하여 회귀를 실행하는 방법
나는 이것이 간단하다고 확신하지만 파이썬의 초보자에게는 pandas
데이터 프레임 에서 변수를 반복 하고 각각과 회귀를 실행하는 방법을 알아내는 데 어려움을 겪고 있습니다.
내가하고있는 일은 다음과 같습니다.
all_data = {}
for ticker in ['FIUIX', 'FSAIX', 'FSAVX', 'FSTMX']:
all_data[ticker] = web.get_data_yahoo(ticker, '1/1/2010', '1/1/2015')
prices = DataFrame({tic: data['Adj Close'] for tic, data in all_data.iteritems()})
returns = prices.pct_change()
다음과 같이 회귀를 실행할 수 있다는 것을 알고 있습니다.
regs = sm.OLS(returns.FIUIX,returns.FSTMX).fit()
그러나 데이터 프레임의 각 열에 대해이 작업을 수행하려고한다고 가정합니다. 특히 FSTMX에서 FIUIX를, FSTMX에서 FSAIX, FSTMX에서 FSAVX를 회귀하고 싶습니다. 각 회귀 후 잔차를 저장하고 싶습니다.
나는 다음의 다양한 버전을 시도했지만 구문이 잘못되어 있어야합니다.
resids = {}
for k in returns.keys():
reg = sm.OLS(returns[k],returns.FSTMX).fit()
resids[k] = reg.resid
문제는 키로 리턴 열을 참조하는 방법을 모른다 returns[k]
는 것이므로 아마도 잘못된 것입니다.
이를 수행하는 가장 좋은 방법에 대한 지침은 대단히 감사하겠습니다. 아마도 내가 놓친 일반적인 팬더 접근법이있을 것입니다.
for column in df:
print(df[column])
당신은 사용할 수 있습니다 iteritems()
:
for name, values in df.iteritems():
print('{name}: {value}'.format(name=name, value=values[0]))
이 답변은 선택한 열과 DF의 모든 열 을 반복 하는 것입니다.
df.columns
DF의 모든 열 이름을 포함하는 목록을 제공합니다. 이제 모든 열을 반복하려는 경우별로 도움이되지 않습니다. 그러나 선택한 열만 반복하려고 할 때 유용합니다.
파이썬의 목록 슬라이싱을 사용하여 필요에 따라 df.columns를 쉽게 슬라이스 할 수 있습니다. 예를 들어 첫 번째 열을 제외한 모든 열을 반복하려면 다음을 수행하십시오.
for column in df.columns[1:]:
print(df[column])
모든 열을 역순으로 반복하는 것과 마찬가지로 다음과 같이 할 수 있습니다.
for column in df.columns[::-1]:
print(df[column])
이 기법을 사용하여 모든 열을 여러 가지 멋진 방법으로 반복 할 수 있습니다. 또한 다음을 사용하여 모든 열의 인덱스를 쉽게 얻을 수 있습니다.
for ind, column in enumerate(df.columns):
print(ind, column)
을 사용하여 위치별로 데이터 프레임 열을 색인 할 수 있습니다 ix
.
df1.ix[:,1]
예를 들어 첫 번째 열을 반환합니다. (0은 색인 일 것입니다)
df1.ix[0,]
첫 번째 행을 반환합니다.
df1.ix[:,1]
이것은 행 0과 열 1의 교차점에있는 값입니다.
df1.ix[0,1]
등등. 따라서 enumerate()
returns.keys():
숫자를 사용하여 데이터 프레임을 인덱싱 할 수 있습니다 .
해결 방법은 DataFrame
행을 바꾸고 반복하는 것입니다.
for column_name, column in df.transpose().iterrows():
print column_name
목록 이해를 사용하면 모든 열 이름 (헤더)을 얻을 수 있습니다.
[column for column in df]
조금 늦었지만 여기에 내가 한 일이 있습니다. 단계:
- 모든 열 목록 만들기
- itertools를 사용하여 x 조합을 취하십시오
- 제외 된 열 목록과 함께 각 결과 R 제곱 값을 결과 데이터 프레임에 추가
- Sort the result DF in descending order of R squared to see which is the best fit.
This is the code I used on DataFrame called aft_tmt
. Feel free to extrapolate to your use case..
import pandas as pd
# setting options to print without truncating output
pd.set_option('display.max_columns', None)
pd.set_option('display.max_colwidth', None)
import statsmodels.formula.api as smf
import itertools
# This section gets the column names of the DF and removes some columns which I don't want to use as predictors.
itercols = aft_tmt.columns.tolist()
itercols.remove("sc97")
itercols.remove("sc")
itercols.remove("grc")
itercols.remove("grc97")
print itercols
len(itercols)
# results DF
regression_res = pd.DataFrame(columns = ["Rsq", "predictors", "excluded"])
# excluded cols
exc = []
# change 9 to the number of columns you want to combine from N columns.
#Possibly run an outer loop from 0 to N/2?
for x in itertools.combinations(itercols, 9):
lmstr = "+".join(x)
m = smf.ols(formula = "sc ~ " + lmstr, data = aft_tmt)
f = m.fit()
exc = [item for item in x if item not in itercols]
regression_res = regression_res.append(pd.DataFrame([[f.rsquared, lmstr, "+".join([y for y in itercols if y not in list(x)])]], columns = ["Rsq", "predictors", "excluded"]))
regression_res.sort_values(by="Rsq", ascending = False)
Based on the accepted answer, if an index corresponding to each column is also desired:
for i, column in enumerate(df):
print i, df[column]
The above df[column]
type is Series
, which can simply be converted into numpy
ndarray
s:
for i, column in enumerate(df):
print i, np.asarray(df[column])
To iterate over the rows of a data frame (rather than its column names as mostly shown in the other answers), you can use
# df has 3 columns and 5 rows
df = pd.DataFrame(np.random.randint(0, 10, (5, 3)), columns=['A','B','C'])
for col in df.values:
print(col)
which outputs
[5 5 0]
[7 4 5]
[4 1 6]
[2 3 4]
[6 0 4]
To iterate by column rather than by row, simply transpose df.values
:
for col in df.values.T:
print(col)
[5 7 4 2 6]
[5 4 1 3 0]
[0 5 6 4 4]
'Programing' 카테고리의 다른 글
프로그래밍 방식으로 축소 또는 축소 축소 도구 모음 레이아웃 (0) | 2020.06.10 |
---|---|
Clojure에 목록에 특정 값이 포함되어 있는지 테스트 (0) | 2020.06.09 |
왜 파이썬 3에서 부동 소수점 값 4 * 0.1이 멋지게 보이지만 3 * 0.1은 그렇지 않습니까? (0) | 2020.06.09 |
Spring에서 Tomcat이 제공하는 JNDI DataSource를 사용하는 방법은 무엇입니까? (0) | 2020.06.09 |
노드 : 콘솔 대신 파일에 로그인 (0) | 2020.06.09 |