[DF913] 02. 명령 인자로 숫자의 총합 구하기

02. 명령 인자로 숫자의 총합 구하기

http://www.dafit.me/question/?q=YToxOntzOjEyOiJrZXl3b3JkX3R5cGUiO3M6MzoiYWxsIjt9&bmode=view&idx=2968780&t=board

[DF913] 파이썬 기초 훈련 10제 : 다핏, 데이터사이언스 연습문제

 

처음에 보고 add_all 함수는 만들기 쉽네...라고 생각했다가 (단, ~~)을 보고 살짝 당황했던 문제다. 

동아리에서 시스템해킹 스터디를 들을 때 exploit을 배우면서 gdb에서 python 실행하는 법을 같이 배워서 그나마 금방 

'add_all.py ‒c 1 2 3'이런 실행 방식을 이해할 수 있었다. 이래서 대충 공부하면 안 되는 건가...

 

(1)

이 부분은 저번 학기 때 들었던 내용이 많은 도움이 되었다. *args와 **kwargs에 대한 부분이었는데 배울 땐 이해하느라 고생했지만 배우고 나니 쓸모가 많다.

 

아래 블로그가 내용을 잘 정리한 것 같아 첨부한다.

https://brunch.co.kr/@princox/180

[나름 중급 파이썬1] *args와 **kwargs

그래서 내 풀이는 다음과 같다.

def add_all(*arg):
    sum = 0
    for i in arg:
        sum += i
    return sum

몇 개의 인자를 받을지 불확실하니 *을 붙여서 들어오는 모든 인자들을 튜플 형태로 arg에 저장한다.

파이썬의 편한 반복문! for in을 이용해서 arg의 모든 인자들을 더해준다. 그리고 리턴한다. 끝!

add_all(1, 2, 3)

6

add_all(1, 2, 3, 4, 5, 6, 7, 8, 9, 10)

55

잘 작동한다.

 

(2)

문제의 Cmd환경!

시스템 해킹에서 공부할 때는 c로 쓰인 exe파일을 python으로 입력을 줬던 거라 python파일에서는 어떻게 명령 인자가 전달되는지 잘 모르겠어서 열심히 검색을 해봤다.

 

아래 블로그에서 도움을 받았다.

https://brownbears.tistory.com/202

[Python] 커맨드라인 인자값 (스크립트 파라미터) 구하기

import sys
for i in range(1, len(sys.argv)):
	print(sys.argv[i])

system모듈을 이용해서 받은 입력들을 하나씩 다 출력해주는 구문인데 여기서 포인트는 범위가 1부터 시작한다는 거다.

왜냐하면 실행할 때,

 

python 파일 이름. py ~~ 명령 인자

 

를 cmd 창에 입력해서 실행하는데 저기 저 '파일 이름. py'부터 튜플에 저장이 되기 때문이다. 즉, 0번째 부분에 파일 이름이 들어가 버린다. 그래서 꼭 제외를 해줘야 한다.

 

import sys
sum=0
for i, j in enumerate(sys.argv):
    if(i!=0 and i!=1):   #add_all.py ‒c 을 제외해 주는 구문
        sum+=int(j)
print(sum)

나는 enumerate를 이용해서 튜플에 번호를 새겨줘서 번호가 0과 1일 땐 더하지 않고 넘어가도록 만들었는데 지금 생각해 보니까 range(2, sys.argv)로 했어도 괜찮았었을 것 같다.

여기서 시작 부분이 2인 이유는 우리의 입력은 python add_all.py -c 1 2 3 이여서 add_all.py와 -c 두 개를 넘겨줘야 하기 때문이다.

cmd 환경에서도 잘 작동하는 모습을 확인할 수 있다.

---