[Python] argparse

파이썬 스크립트를 개발할 때, 호출 당시 인자값을 줘서 동작을 다르게 하고 싶은 경우가 있다.

이때, 파이썬 내장함수인 argparse 모듈을 사용하여 프로그램을 실행시에 커맨드 라인에 인수를 받아 처리를 간단히 할 수 있다.

 

# 1. argparse를 import
import argparse

# 2. 인자값을 받을 수 있는 인스턴스 생성
p = argparse.ArgumentParser()

# 3. 입력받을 인자값 등록
p.add_argument('--model_fn', required=True)
p.add_argument('--train_ratio', type=float, default=.8)
p.add_argument('--batch_size', type=int, default=64)
p.add_argument('--n_epochs', type=int, default=20)
p.add_argument('--verbose', type=int, default=2)

# 입력받은 인자값을 config에 저장
config = p.parse_args()

# 출력 테스트
print(config.model_fn)
print(config.train_ratio)
print(config.batch_size)
print(config.n_epochs)
print(config.verbose)

◾ 그냥 실행 결과

 

◾ 인자값을 주고 실행시킨 결과