[python] argparse.ArgumentParser() 사용법

기본적으로 ArgumentParser는 프로그램 실행 시, 실행시킬 파일의 파일명이나 파라미터를 지정할 수 있는 기능을 가진 라이브러리 이다. 

단계별로 사용법을 알아보자.

 

1. 라이브러리 임포트

import argparse

2. 전달 인자(=파일명 또는 파라미터)를 저장할 객체 선언

parser = argparse.ArgumentParser()

 

3. 객체에 인자 지정

    - 이때 다양한 option이 존재하기 때문에 그때그때 찾아쓰는걸 권장한다.

    - 일반적으로 다음과 같은 예시에서는 -d 는 인자 이름, type은 문자열, default 값은 movie, help란 인자에 대한 description을 의미한다.

parser.add_argument('-d', type=str, default='movie', help='which dataset to preprocess')

 

4. 객체를 새로운 변수의 이름으로 할당. 

args = parser.parse_args()

 

5. arg의 d 이름을 가진 인자는 dataset에 할당된다.

DATASET = args.d

 

 

단, 해당 인자들은 cmd 창에서 python 실행 시 인자 값들을 같이 전달해주어야 한다.

(본 예시는 visual studio code에서 실행 함)

$ python preprocess.py -d movie

파일 실행시 인자값을 전달하는 것을 볼 수 있다.