[Python] argparse 사용방법 (커맨드에서 인자로 변수 받아오기)

특정 python 파일을 실행 시킬 때, 특정 변수의 값을 입력 받고 싶을 때 사용하는 방법이다.

입력함수 input()을 통해 받을 수 있고, 따로 config.py 파일을 만들어서 변수를 변경하는 방법도 있지만

파일을 실행 시킬 때 간단하게 입력 가능하기 때문에 많이 쓰이는 것 같다.

 

사용 방법

import argparse

parser = argparse.ArgumentParser()
# 처음으로 인스턴스를 생성 시켜줘야 한다.

parser.add_argument('--model_dir', type=str, default='./model',
                    help='Base directory for the model.', required=False)
# add_argument()를 통해서 인자를 입력 받는다.

args = parser.parse_args()
# args = parser.parse_known_args()
# 입력받은 인자를 저장(type: namespace)

 

$ pip install argparse
# 패키지 다운로드

$ python test.py --model_dir=./model/weight.h5

$ python test.py [--moedel_dir DIR]

$ python test.py -h
# -h를 하게 되면 리눅스 bash 명령어와 같이 인자로 받을 수 있는 옵션과 사용법이 나온다.