[Python] 반복문에서 변수 선언하는 법

 

 

 

데이터 분석을 하다보면 가끔 for문 안에서 계산되는 값들을 각각의 변수에 담고 싶을 때가 있다. (나만 그런가?)

그때, 쓸 수 있는 방법이다.

 

for i in range():
    globals()['변수명'+str(i)] = 계산식

또는

for i in range():
    locals()['변수명'+str(i)] = 계산식

 

globals() 함수는 전역 변수들을 딕셔너리 형태로, locals() 함수는 지역 변수들을 딕셔너리 형태로 반환한다.

'globals()['변수명']  = x' / 'locals()['변수명'] = x' 을 이용하면 해당 딕셔너리에 키와 값을 추가할 수 있다.

필요에 맞게 전역(globals) / 지역(locals)을 골라 쓰면 될 것 같다.

 

아래는 예시이다.

for i in range(1, 5):
    globals()['var_'+str(i)] = i + 3
    
    
var_1
############ 출력결과 : 4

var_2
############ 출력결과 : 5

var_4
############# 출력결과 : 7

 

var_1, var_2, var_3, var_4에 각각 i+3의 값이 담겼다.