회귀) subplot 을 선형회귀 직선 표현(p.303_구판)

안녕하세요!
선생님 강의 잘 듣고 있습니다.

회귀 파트 학습을 하면서 코드 실행을 하는데 하기의 코드 부분이 이해가 되지 않습니다..

fig, axs = plt.subplots(figsize =(16,8), ncols=4, nrows=2)

lm_features = ['RM','ZN','INDUS','NOX','AGE','PTRATIO','LSTAT','RAD']

for i, feature in enumerate(lm_features):

row = int(i/4)

col = i%4

sns.regplot(x=feature, y='PRICE',data=bostonDF, ax = axs[row][col])

* row , col 을 각각 정의한 부분이 이해가 어렵습니다. int(i/4), i%4 가 각각 어떤식으로 형성되는가입니다.

더하여 enumerate 함수만 상기 for 구문에 쓰일 수 있는지도 궁금합니다.(for in 구문은 안되는지)

안녕하십니까,

위 코드는 8개의 컬럼들에 대해서 개별적인 regression plot을 2행 x 4열 grid 선상에 표현하기 위함입니다.

먼저 for 문에 사용된 enumerate()에 대해서 말씀드리겠습니다. enumerate(list)는 인자로 iterable 한 객체, 예를 들어 list([]) 객체를 인자로 받아서, iterate loop 횟수와 list내의 개별 값을 반환합니다.

for i, feature in enumerate(lm_features):

print(i, feature)

row = int(i/4)

col = i%4

의 코드라면 아래와 같은 값을 반환하게 됩니다.

0, RM

1, ZN

2, INDUS

....

7, RAD

즉 i는 ['RM','ZN','INDUS','NOX','AGE','PTRATIO','LSTAT','RAD'] 8개의 원소가 있는 list에 대해서 for for loop indexing 값 0 ~ 7 까지 8개의 숫자값을 차례로 가지게 되는 것입니다.

그리고 row는 2행 x 4열 grid 그림판의 행 위치를 나타냅니다. 2행이므로 0~1 까지의 index 값을 의미합니다. col은 4열이므로 0~3 까지의 index값을 의미합니다.

i가 0 일 때는 row는 0이며 col역시 0 입니다. 따라서 sns.regplot(x=feature, y='PRICE',data=bostonDF, ax = axs[row][col]) 에서 axs[row][col]은 axs[0][0] 이므로 1행 1열에 RM 컬럼의 regplot을 그립니다.

i가 1일 때는 row = int(i/4) 이므로 소수점은 모두 int 변환하여 0이 되므로 row는 역시 0이나 col은 i%4 이므로 1입니다. 따라서 axs[0][1]이고 1행 2열에 ZN의 regplot을 그리게 됩니다.

i가 3일때까지 row = int(i/4) 이므로 소수점은 모두 int 변환하여 0이 되므로 row는 0이며, col만 계속 변합니다.

i가 4가 되면 비로서 row가 1로 바뀝니다. i가 4면 row = int(4/4) 이므로 1이 됩니다. col은 4%4 이므로 0 이 됩니다. 즉 2행 1열에 AGE의 regplot을 그립니다.

동일하게 i가 7까지 계속 수행해 보면 row는 1에서 계속 고정되며 col은 계속 변경됩니다.

이처럼 row = int(i/4), col = i%4 로 하면 4라는 고정된 컬럼 갯수 기반으로 동적으로 할당될 grid의 행과 열을 계산해 줄 수 있습니다.

감사합니다.

인프런 커뮤니티 질문&답변