함수란 무엇인가?
함수를 설명하기 전에 믹서기를 생각해 보자. 우리는 믹서기에 과일을 넣는다. 그리고 믹서기를 이용해서 과일을 갈아 과일 주스를 만든다. 우리가 믹서기에 넣는 과일은 "입력"이 되고 과일 주스는 "출력(결과값)"이 된다.
그렇다면 믹서기는 무엇인가?
(믹서기는 과일을 입력받아 주스를 출력하는 함수와 같다.)
우리가 배우려는 함수가 바로 믹서기와 비슷하다. 입력값을 가지고 어떤 일을 수행한 다음에 그 결과물을 내어놓는 것, 이것이 바로 함수가 하는 일이다. 우리는 어려서부터 함수에 대해 공부했지만 함수에 관해 깊이 생각해 본 적은 별로 없다. 예를 들어 y = 2x + 3도 함수다. 하지만 이를 수학 시간에 배운 직선 그래프로만 알고 있지 x에 어떤 값을 넣었을 때 어떤 변화에의해서 y 값이 나오는지 그 과정에 대해서는 별로 관심을 두지 않았을 것이다.
이제 우리는 함수에 대해서 조금 더 생각해 보는 시간을 가져야 한다. 프로그래밍에 있어서 함수는 정말 중요하기 때문이다. 자, 이제 파이썬 함수의 세계로 깊이 들어가 보자.
함수를 사용하는 이유는 무엇일까?
프로그래밍을 하다 보면 똑같은 내용을 반복해서 작성하고 있는 자신을 발견할 때가 종종 있다. 이때가 바로 함수가 필요한 때이다. 즉, 반복되는 부분이 있을 경우 "반복적으로 사용되는 가치 있는 부분"을 한 뭉치로 묶어서 "어떤 입력값을 주었을 때 어떤 결과값을 돌려준다"라는식의 함수로 작성하는 것이 현명하다.
함수를 사용하는 또 다른 이유는 자신이 만든 프로그램을 함수화하면 프로그램의 흐름을 일목요연하게 볼 수 있기 때문이다. 마치 공장에서 원재료가 여러 공정을 거쳐 하나의 상품이되는 것처럼 프로그램에서도 입력한 값이 여러 함수들을 거치면서 원하는 결과값을 내는 것을 볼 수 있다. 이렇게 되면 프로그램의 흐름도 잘 파악할 수 있고 에러가 어디에서 나는지도 금방 알아차릴 수 있다. 함수를 잘 이용하고 함수를 적절하게 만들 줄 아는 사람이 능력 있는 프로그래머이다.
파이썬 함수의 구조
파이썬 함수의 구조는 다음과 같다.
def 함수명(매개변수):
<수행할 문장1>
<수행할 문장2>
...
def는 함수를 만들 때 사용하는 예약어이며, 함수명은 함수를 만드는 사람이 임의로 만들 수 있다. 함수명 뒤 괄호 안의 매개변수는 이 함수에 입력으로 전달되는 값을 받는 변수이다. 이렇게 함수를 정의한 다음 if, while, for문 등과 마찬가지로 함수에서 수행할 문장들을 입력한다.
간단하지만 많은 것을 설명해 주는 다음의 예를 보자.
def sum(a, b):
return a + b
위 함수는 다음과 같이 풀이된다.
"이 함수의 이름(함수명)은 sum이고 입력으로 2개의 값을 받으며 결과값은 2개의 입력값을 더한 값이다."
여기서 return은 함수의 결과값을 돌려주는 명령어이다. 먼저 다음과 같이 sum 함수를 만들자.
>>> def sum(a, b):
... return a+b
...
>>>
이제 직접 sum 함수를 사용해 보자.
>>> a = 3
>>> b = 4
>>> c = sum(a, b)
>>> print(c)
7
변수 a에 3, b에 4를 대입한 다음 앞서 만든 sum 함수에 a와 b를 입력값으로 넣어 준다. 그리고 변수 c에 sum 함수의 결과값을 대입하면 print(c)로 c의 값을 확인할 수 있다.
매개변수와 인수
매개변수(parameter)와 인수(arguments)는 혼용해서 사용되는 헷갈리는 용어이므로 잘 기억해 두기로 하자. 매개변수는 함수에 입력으로 전달된 값을 받는 변수를 의미하고 인수는 함수를 호출할 때 전달하는 입력값을 의미한다.
def sum(a, b): # a, b는 매개변수
return a+b
print(sum(3, 4)) # 3, 4는 인수
[같은 의미를 가진 여러 가지 용어들에 주의하자]
프로그래밍을 공부할 때 어려운 부분 중 하나가 용어의 혼용이라고 할 수 있다. 우리는 공부하면서 원서를 보기도 하고 누군가의 번역본을 보기도 하면서 의미는 같지만 표현이 다른 용어들을 자주 만나게 된다. 한 예로 입력값을 다른 말로 함수의 인수, 입력 인수 등으로 말하기도 하고 결과값을 출력값, 리턴값, 돌려주는 값 등으로 말하기도 한다. 이렇듯 많은 용어들이 여러 가지 다른 말로 표현되지만 의미는 동일한 경우가 많다. 이런 것들을 기억해 놓아야 머리가 덜 아플 것이다.
입력값과 결과값에 따른 함수의 형태
함수는 들어온 입력값을 받아 어떤 처리를 하여 적절한 결과값을 돌려준다.
입력값 ---> 함수 ----> 리턴값
함수의 형태는 입력값과 결과값의 존재 유무에 따라 4가지 유형으로 나뉜다. 자세히 알아보자.
일반적인 함수
입력값이 있고 결과값이 있는 함수가 일반적인 함수이다. 앞으로 여러분이 프로그래밍을 할 때 만들 함수는 대부분 다음과 비슷한 형태일 것이다.
def 함수이름(매개변수):
<수행할 문장>
...
return 결과값
다음은 일반적인 함수의 전형적인 예이다.
def sum(a, b):
result = a + b
return result
sum 함수는 2개의 입력값을 받아서 서로 더한 결과값을 돌려준다.
이 함수를 사용하는 방법은 다음과 같다. 입력값으로 3과 4를 주고 결과값을 돌려받아 보자.
>>> a = sum(3, 4)
>>> print(a)
7
이처럼 입력값과 결과값이 있는 함수의 사용법을 정리하면 다음과 같다.
결과값을 받을 변수 = 함수명(입력 인수 1, 입력 인수 2, ...)
입력값이 없는 함수
입력값이 없는 함수가 존재할까? 당연히 존재한다. 다음을 보자.
>>> def say():
... return 'Hi'
...
>>>
say라는 이름의 함수를 만들었다. 그런데 매개변수 부분을 나타내는 함수명 뒤의 괄호 안이 비어 있다. 이 함수는 어떻게 사용하는 걸까?
다음을 직접 입력해 보자.
>>> a = say()
>>> print(a)
Hi
위의 함수를 쓰기 위해서는 say()처럼 괄호 안에 아무런 값도 넣지 않아야 한다. 이 함수는 입력값은 없지만 결과값으로 Hi라는 문자열을 돌려준다. a = say()처럼 작성하면 a에 Hi라는 문자열이 대입되는 것이다.
이처럼 입력값이 없고 결과값만 있는 함수는 다음과 같이 사용된다.
결과값을 받을 변수 = 함수명()
결과값이 없는 함수
결과값이 없는 함수 역시 존재한다. 다음의 예를 보자.
>>> def sum(a, b):
... print("%d, %d의 합은 %d입니다." % (a, b, a+b))
...
>>>
결과값이 없는 함수는 호출해도 돌려주는 값이 없기 때문에 다음과 같이 사용한다.
>>> sum(3, 4)
3, 4의 합은 7입니다.
즉, 결과값이 없는 함수는 다음과 같이 사용한다.
함수명(입력 인수1, 입력 인수2, ...)
결과값이 진짜 없는지 확인하기 위해 다음의 예를 직접 입력해 보자.
>>> a = sum(3, 4)
3, 4의 합은 7입니다.
아마도 여러분은 "3, 4의 합은 7입니다."라는 문장을 출력해 주었는데 왜 결과값이 없다는 것인지 의아하게 생각할 것이다. 이 부분이 초보자들이 혼란스러워하는 부분이기도 한데 print문은 함수의 구성 요소 중 하나인 <수행할 문장>에 해당하는 부분일 뿐이다. 결과값은 당연히 없다. 결과값은 오직 return 명령어로만 돌려받을 수 있다.
이를 확인해 보자. 돌려받을 값을 a라는 변수에 대입하여 출력해 보면 결과값이 있는지 없는지 알 수 있다.
>>> a = sum(3, 4)
>>> print(a)
None
a의 값은 None이다. None이란 거짓을 나타내는 자료형이라고 언급한 적이 있다. sum 함수처럼 결과값이 없을 때 a = sum(3, 4)처럼 쓰게 되면 함수 sum은 리턴값으로 a 변수에 None을 돌려준다. 이것을 가지고 결과값이 있다고 생각하면 곤란하다.
입력값도 결과값도 없는 함수
입력값도 결과값도 없는 함수 역시 존재한다. 다음의 예를 보자.
>>> def say():
... print('Hi')
...
>>>
입력 인수를 받는 매개변수도 없고 return문도 없으니 입력값도 결과값도 없는 함수이다. 이 함수를 사용하는 방법은 단 한 가지이다.
>>> say()
Hi
즉, 입력값도 결과값도 없는 함수는 다음과 같이 사용한다.
함수명()
매개변수 지정하여 호출하기
함수를 호출 할 때 매개변수를 지정하여 호출할 수도 있다. 다음의 예를 보자.
>>> def sum(a, b):
... return a+b
...
앞서 알아보았던 sum함수이다. 이 함수를 우리는 다음과 같이 매개변수를 지정하여 사용할 수 있다.
>>> sum(a=3, b=7) # a에 3, b에 7을 전달
10
매개변수를 지정하면 다음과 같이 순서에 상관없이 사용할 수 있다는 장점이 있다.
>>> sum(b=5, a=3) # b에 5, a에 3을 전달
8
입력값이 몇 개가 될지 모를 때는 어떻게 해야 할까?
입력값이 여러 개일 때 그 입력값들을 모두 더해 주는 함수를 생각해 보자. 하지만 몇 개가 입력될지 모를 때는 어떻게 해야 할까? 아마도 난감할 것이다. 파이썬은 이런 문제를 해결하기 위해 다음과 같은 방법을 제공한다.
def 함수이름(*매개변수):
<수행할 문장>
...
일반적으로 볼 수 있는 함수 형태에서 괄호 안의 매개변수 부분이 *매개변수로 바뀌었다.
여러 개의 입력값을 받는 함수 만들기
다음의 예를 통해 여러 개의 입력값을 모두 더하는 함수를 직접 만들어 보자. 예를 들어 sum_many(1, 2)이면 3을, sum_many(1,2,3)이면 6을, sum_many(1, 2, 3, 4, 5, 6, 7, 8, 9, 10)이면 55를 돌려주는 함수를 만들어 보자.
>>> def sum_many(*args):
... sum = 0
... for i in args:
... sum = sum + i
... return sum
...
>>>
위에서 만든 sum_many라는 함수는 입력값이 몇 개이든 상관이 없다. *args처럼 매개변수명 앞에 *을 붙이면 입력값들을 전부 모아서 튜플로 만들어 주기 때문이다. 만약 sum_many(1, 2, 3)처럼 이 함수를 쓰면 args는 (1, 2, 3)이 되고, sum_many(1, 2, 3, 4, 5, 6, 7, 8, 9, 10)처럼 쓰면 args는 (1, 2, 3, 4, 5, 6, 7, 8, 9, 10)이 된다. 여기서 *args라는 것은 임의로정한 변수명이다. *pey, *python처럼 아무 이름이나 써도 된다.
※ args는 입력 인수를 뜻하는 영어 단어인 arguments의 약자이며 관례적으로 자주 사용한다.
실제로 이 함수를 직접 입력해 보자.
>>> result = sum_many(1,2,3)
>>> print(result)
6
>>> result = sum_many(1,2,3,4,5,6,7,8,9,10)
>>> print(result)
55
sum_many(1,2,3)으로 함수를 호출하면 6을 리턴하고, sum_many(1, 2, 3, 4, 5, 6, 7, 8, 9, 10)을 대입하면 55를 리턴한다.
def sum_many(*args)처럼 함수의 인수로 *args(*매개변수)만 사용할 수 있는 것은 아니다. 다음의 예를 보자.
>>> def sum_mul(choice, *args):
... if choice == "sum":
... result = 0
... for i in args:
... result = result + i
... elif choice == "mul":
... result = 1
... for i in args:
... result = result * i
... return result
...
>>>
위의 예는 매개변수로 choice와 *args를 사용했다. 따라서 다음과 같이 쓸 수 있다.
즉, sum_mul('sum', 1,2,3,4) 또는 sum_mul('mul', 1,2,3,4,5)처럼 choice 부분에 'sum'이나 'mul'이라는 문자열을 입력값으로 준다. 그리고 그 뒤에는 개수에 상관없이 여러 개의 숫자를 입력값으로 준다.
실제로 한 번 테스트해 보자.
>>> result = sum_mul('sum', 1,2,3,4,5)
>>> print(result)
15
>>> result = sum_mul('mul', 1,2,3,4,5)
>>> print(result)
120
첫 번째 입력 인수에 'sum'이 입력된 경우 *args에 입력되는 모든 값을 더해서 15를 돌려주고, 'mul'이 입력된 경우 *args에 입력되는 모든 값을 곱해서 120을 돌려준다.
키워드 파라미터 kwargs
이번에는 키워드 파라미터인 **kwargs에 대해서 알아보자. kwargs는 keyword arguments의 약어이다. **kwargs는 *args와는 달리 별표시(*)가 두 개 사용된다. 역시 이것도 예제로 알아보도록 하자.
먼저 다음과 같은 함수를 작성 해 보자.
>>> def func(**kwargs):
... print(kwargs)
...
그리고 이 함수를 다음과 같이 사용해 보자.
>>> func(a=1)
{'a': 1}
>>> func(name='foo', age=3)
{'age': 3, 'name': 'foo'}
func 함수의 인수로 key=value 형태를 주었을 때 입력 값 전체가 kwargs라는 딕셔너리에 저장된다는 것을 알 수 있다.
즉, **kwargs는 모든 key=value 형태의 입력인수가 저장되는 딕셔너리 변수이다.
이번에는 다음과 같은 형태의 호출에 대해서 생각해 보자.
>>> func(1, 2, 3, name='foo', age=3)
위 예처럼 이렇게 입력인수의 형태가 다양한 경우 입력인수의 갯수에 상관없이 입력을 받고 싶다면 어떻게 해야 할까?
이런경우 다음의 함수를 사용하면 입력인수를 모두 처리할 수 있게 된다.
>>> def func(*args, **kwargs):
... print(args)
... print(kwargs)
...
>>> func(1, 2, 3, name='foo', age=3)
(1, 2, 3)
{'age': 3, 'name': 'foo'}
1, 2, 3 과 같은 일반적인 입력은 args튜플에 저장되고 name='foo' 와 같은 형태의 입력은 kwargs 딕셔너리에 저장되는 것을 확인할 수 있다.
만약 다음과 같이 호출한다면 어떻게 될까?
>>> func(1, 2, 3, name='foo', age=3, 4, 5)
일반적인 입력인수가 키워드형태의 입력 뒤에 존재하는 경우이다. 호출하면 다음과 같은 오류를 만나게 된다.
>>> func(1, 2, 3, name='foo', age=3, 4, 5)
File "<stdin>", line 1
SyntaxError: non-keyword arg after keyword arg
키워드 형태의 인수뒤에 키워드 형태가 아닌 인수는 올 수 없음을 알 수 있다. 따라서 다음과 같은 형태의 호출로 변경해야 할 것이다.
>>> func(1, 2, 3, 4, 5, name='foo', age=3)
(1, 2, 3, 4, 5)
{'age': 3, 'name': 'foo'}
함수의 결과값은 언제나 하나이다
먼저 다음의 함수를 만들어 보자.
>>> def sum_and_mul(a,b):
... return a+b, a*b
※ sum_and_mul 함수는 2개의 입력 인수를 받아 더한 값과 곱한 값을 돌려주는 함수이다.
이 함수를 다음과 같이 호출하면 어떻게 될까?
>>> result = sum_and_mul(3,4)
결과값은 a+b와 a*b 2개인데 결과값을 받아들이는 변수는 result 하나만 쓰였으니 오류가 발생하지 않을까? 당연한 의문이다. 하지만 오류는 발생하지 않는다. 그 이유는 함수의 결과값은 2개가 아니라 언제나 1개라는 데 있다. sum_and_mul 함수의 결과값 a+b와 a*b는 튜플값 하나인 (a+b, a*b)로 돌려준다.
따라서 result 변수는 다음과 같은 값을 갖게 된다.
result = (7, 12)
즉, 결과값으로 (7, 12)라는 튜플 값을 갖게 되는 것이다.
만약 이 하나의 튜플값을 2개의 결과값처럼 받고 싶다면 다음과 같이 함수를 호출하면 된다.
>>> sum, mul = sum_and_mul(3, 4)
이렇게 호출하면 sum, mul = (7, 12)가 되어 sum은 7이 되고 mul은 12가 된다.
또 다음과 같은 의문이 생길 수도 있다.
>>> def sum_and_mul(a,b):
... return a+b
... return a*b
...
>>>
위와 같이 return문을 2번 사용하면 2개의 결과값을 돌려주지 않을까? 하지만 파이썬에서 위와 같은 함수는 참 어리석은 함수이다.
그 이유는 sum_and_mul 함수를 호출해 보면 알 수 있다.
>>> result = sum_and_mul(2, 3)
>>> print(result)
5
※ sum_and_mul(2, 3)의 결과값은 5 하나뿐이다. 두 번째 return문인
return a*b는 실행되지 않았다는 뜻이다.
이 예에서 볼 수 있듯이 두 번째 return문인 return a*b는 실행되지 않았다. 따라서 이 함수는 다음과 완전히 동일하다.
>>> def sum_and_mul(a,b):
... return a+b
...
>>>
즉, 함수는 return문을 만나는 순간 결과값을 돌려준 다음 함수를 빠져나가게 된다.
[return의 또 다른 쓰임새]
어떤 특별한 상황이 되면 함수를 빠져나가고자 할 때는 return을 단독으로 써서 함수를 즉시 빠져나갈 수 있다. 다음 예를 보자.
>>> def say_nick(nick):
... if nick == "바보":
... return
... print("나의 별명은 %s 입니다." % nick)
...
>>>
위의 함수는 "별명"을 입력으로 전달받아 출력하는 함수이다. 이 함수 역시 리턴값은 없다(문자열을 출력한다는 것과 리턴값이 있다는 것은 전혀 다른 말이다. 혼동하지 말자. 함수의 리턴값은 오로지 return문에 의해서만 생성된다).
만약에 입력값으로 '바보'라는 값이 들어오면 문자열을 출력하지 않고 함수를 즉시 빠져나간다.
>>> say_nick('야호')
나의 별명은 야호입니다.
>>> say_nick('바보')
>>>
이처럼 return으로 함수를 빠져나가는 방법은 실제 프로그래밍에서 자주 사용된다.
매개변수에 초깃값 미리 설정하기
이번에는 조금 다른 형태로 함수의 인수를 전달하는 방법에 대해서 알아보자. 매개변수에 초깃값을 미리 설정해 주는 경우이다.
※ 앞으로 나올 프로그램 소스에서 >>> 표시가없으면 에디터로 작성하기를 권장한다는 뜻이다.
def say_myself(name, old, man=True):
print("나의 이름은 %s 입니다." % name)
print("나이는 %d살입니다." % old)
if man:
print("남자입니다.")
else:
print("여자입니다.")
※ say_myself 함수는 3개의 입력 인수를 받아서 마지막 인수인 man이 True이면 "남자입니다.", False이면 "여자입니다."를 출력한다.
위의 함수를 보면 매개변수가 name, old, man=True 이렇게 3개다. 그런데 낯선 것이 나왔다. man=True처럼 매개변수에 미리 값을 넣어 준 것이다. 이것이 바로 함수의 매개변수 초깃값을 설정하는 방법이다. 함수의 매개변수에 전달되는 값이 항상 변하는 것이 아닐 경우에는 이렇게 함수의 초깃값을 미리 설정해 두면 유용하다.
say_myself 함수는 다음처럼 사용할 수 있다.
say_myself("박응용", 27)
say_myself("박응용", 27, True)
입력값으로 "박응용", 27처럼 2개를 주면 name에는 "박응용"이 old에는 27이 전달된다. 그리고 man이라는 변수에는 입력값을 주지 않았지만 초깃값인 True 값을 갖게 된다.
따라서 위의 예에서 함수를 사용한 2가지 방법은 모두 동일한 결과를 출력한다.
나의 이름은 박응용입니다.
나이는 27살입니다.
남자입니다.
이제 초깃값이 설정된 부분을 False로 바꿔 보자.
say_myself("박응선", 27, False)
man 변수에 False 값이 전달되어 다음과 같은 결과가 출력된다.
나의 이름은 박응선입니다.
나이는 27살입니다.
여자입니다.
함수 매개변수에 초깃값을 설정할 때 주의할 사항
함수의 매개변수에 초깃값을 설정할 때 주의할 것이 하나 있다. 만약 위에서 본 say_myself 함수를 다음과 같이 만들면 어떻게 될까?
def say_myself(name, man=True, old):
print("나의 이름은 %s 입니다." % name)
print("나이는 %d살입니다." % old)
if man:
print("남자입니다.")
else:
print("여자입니다.")
이전 함수와 바뀐 부분은 초깃값을 설정한 매개변수의 위치이다. 결론을 미리 말하면 이것은 함수를 실행할 때 오류가 발생한다.
얼핏 생각하기에 위의 함수를 호출하려면 다음과 같이 하면 될 것 같다.
say_myself("박응용", 27)
위와 같이 함수를 호출한다면 name 변수에는 "박응용"이 들어갈 것이다. 하지만 파이썬 인터프리터는 27을 man 변수와 old 변수 중 어느 곳에 전달해야 할지 알 수 없게 된다.
오류 메시지를 보면 다음과 같다.
SyntaxError: non-default argument follows default argument
위의 오류 메시지는 초깃값을 설정해 놓은 매개변수 뒤에 초깃값을 설정해 놓지 않은 매개변수는 사용할 수 없다는 말이다. 즉, 매개변수로 (name, old, man=True)는 되지만 (name, man=True, old)는 안 된다는 것이다. 초기화시키고 싶은 매개변수들을 항상 뒤쪽에 위치시키는 것을 잊지 말자.
함수 안에서 선언된 변수의 효력 범위
함수 안에서 사용할 변수의 이름을 함수 밖에서도 동일하게 사용한다면 어떻게 될까? 이런 궁금증이 생겼던 독자라면 이번에 확실하게 답을 찾을 수 있을 것이다.
아래의 예를 보자.
# vartest.py
a = 1
def vartest(a):
a = a +1
vartest(a)
print(a)
먼저 a라는 변수를 생성하고 1을 대입한다. 다음 입력으로 들어온 값에 1을 더해 주고 결과값은 돌려주지 않는 vartest 함수를 선언한다. 그리고 vartest 함수에 입력값으로 a를 주었다. 마지막으로 a의 값을 출력하는 print(a)를 입력한다. 과연 결과값은 무엇이 나올까?
당연히 vartest 함수에서 매개변수 a의 값에 1을 더했으니까 2가 출력되어야 할 것 같지만 프로그램 소스를 작성해서 실행시켜 보면 결과값은 1이 나온다. 그 이유는 함수 안에서 새로 만들어진 매개변수는 함수 안에서만 사용되는 "함수만의 변수"이기 때문이다. 즉, def vartest(a)에서 입력 값을 전달받는 매개변수 a는 함수 안에서만 사용되는 변수이지 함수 밖의 변수 a가 아니라는 뜻이다.
따라서 vartest 함수는 다음처럼 변수 이름을 hello로 한 vartest 함수와 완전히 동일하다.
def vartest(hello):
hello = hello + 1
즉, 함수 안에서 사용되는 매개변수는 함수 밖의 변수 이름들과는 전혀 상관이 없다는 말이다.
다음의 예를 보면 더욱 분명하게 이해할 수 있을 것이다.
# vartest_error.py
def vartest(a):
a = a + 1
vartest(3)
print(a)
위의 프로그램 소스를 에디터로 작성해서 실행시키면 어떻게 될까? 오류가 발생할 것이라고 생각한 독자는 모든 것을 이해한 독자이다. vartest(3)을 수행하면 vartest라는 함수 내에서 a는 4가 되지만 함수를 호출하고 난 뒤에 print(a)라는 문장은 에러가 발생하게 된다. 그 이유는 print(a)에서 사용된 a라는 변수는 어디에서도 찾을 수가 없기 때문이다. 다시 말하지만 함수 안에서 선언된 매개변수는 함수 안에서만 사용될 뿐 함수 밖에서는 사용되지 않는다. 이것을 이해하는 것은 매우 중요하다.
그렇다면 vartest라는 함수를 이용해서 함수 밖의 변수 a를 1만큼 증가시킬 수 있는 방법은 없을까? 이 질문에는 2가지 해결 방법이 있다.
함수 안에서 함수 밖의 변수를 변경하는 방법
1. return 이용하기
# vartest_return.py
a = 1
def vartest(a):
a = a +1
return a
a = vartest(a)
print(a)
첫 번째 방법은 return을 이용하는 방법이다. vartest 함수는 입력으로 들어온 값에 1을 더한값을 돌려준다. 따라서 a = vartest(a)라고 대입하면 a가 vartest 함수의 결과값으로 바뀐다. 여기서도 물론 vartest 함수 안의 a 매개변수는 함수 밖의 a와는 다른 것이다.
2. global 명령어 이용하기
# vartest_global.py
a = 1
def vartest():
global a
a = a+1
vartest()
print(a)
두 번째 방법은 global이라는 명령어를 이용하는 방법이다. 위의 예에서 볼 수 있듯이 vartest 함수 안의 global a라는 문장은 함수 안에서 함수 밖의 a 변수를 직접 사용하겠다는 뜻이다. 하지만 프로그래밍을 할 때 global 명령어는 사용하지 않는 것이 좋다. 왜냐하면 함수는 독립적으로 존재하는 것이 좋기 때문이다. 외부 변수에 종속적인 함수는 그다지 좋은 함수가 아니다. 그러므로 가급적 global 명령어를 사용하는 이 방법은 피하고 첫 번째 방법을 사용하기를 권한다.
lambda
lambda는 함수를 생성할 때 사용하는 예약어로, def와 동일한 역할을 한다. 보통 함수를 한줄로 간결하게 만들 때 사용한다. 우리말로는 "람다"라고 읽고 def를 사용해야 할 정도로 복잡하지 않거나 def를 사용할 수 없는 곳에 주로 쓰인다. 사용법은 다음과 같다.
lambda 매개변수1, 매개변수2, ... : 매개변수를 이용한 표현식
한번 직접 만들어 보자.
>>> sum = lambda a, b: a+b
>>> sum(3,4)
7
sum은 두개의 인수를 받아 서로 더한 값을 리턴하는 lambda 함수이다. 위의 예제는 def를 사용한 아래 함수와 하는 일이 완전히 동일하다.
>>> def sum(a, b):
... return a+b
...
>>>
그렇다면 def가 있는데 왜 lambda라는 것이 나오게 되었을까? 이유는 간단하다. lambda는 def 보다 간결하게 사용할 수 있기 때문이다. 또한 lambda는 def를 사용할 수 없는 곳에도 사용할 수 있다. 다음 예제에서 리스트 내에 lambda가 들어간 경우를 살펴보자.
>>> myList = [lambda a,b:a+b, lambda a,b:a*b]
>>> myList
[at 0x811eb2c>, at 0x811eb64>]
즉, 리스트 각각의 요소에 lambda 함수를 만들어 바로 사용할 수 있다. 첫 번째 요소 myList[0]은 2개의 입력값을 받아 두 값의 합을 돌려주는 lambda 함수이다.
>>> myList[0]
at 0x811eb2c>
>>> myList[0](3,4)
7
두 번째 요소 myList[1]은 2개의 입력값을 받아 두 값의 곱을 돌려주는 lambda 함수이다.
>>> myList[1](3,4)
12
파이썬에 익숙해질수록 lambda 함수가 굉장히 편리하다는 사실을 알게 될 것이다.
'파이썬(Python) > 파이썬(Python) 자료구조' 카테고리의 다른 글
| 파이썬 패키지 (0) | 2018.08.08 |
|---|---|
| 파이썬 모듈 (0) | 2018.08.08 |
| 파이썬 클래스 (0) | 2018.08.08 |
| 파이썬에서 파일 읽기&쓰기 (0) | 2018.08.07 |
| 파이썬에서 입력과 출력 (0) | 2018.08.07 |
댓글