[코딩 테스트를 위한 파이썬 문법] 함수

본 포스팅은 나동빈 저자의 ‘이것이 취업을 위한 코딩테스트’를 공부하며 정리한 노트입니다.

함수란?

동일한 알고리즘을 반복적으로 수행해야 할 때 함수는 중요하게 사용된다. 함수를 작성할 때는 함수 내부에서 사용되는 변수의 값을 전달받기 위해 매개변수를 정의할 수 있다. 이후에 함수에서 어떠한 값을 반환하고자 할 때는 return을 이용한다. 함수에서 매개변수나 return문은 존재하지 않을 수도 있다.

또한 함수를 호출하는 과정에서 인자를 넘겨줄 때, 파라미터의 변수를 직접 지정해서 값을 넣을 수 있다. 이 경우 매개변수의 순서가 달라도 상관없다.

def add(a, b):
    print(a+b)

add(b = 4, a = 6)

함수 안에서 함수 밖의 변수 데이터를 변경해야 하는 경우

함수에서 global 키워드를 이용하면 된다. global 키워드로 변수를 지정하면, 해당 함수에서 지역 변수를 만들지 않고, 함수 바깥에 선언된 변수를 바로 참조한다.

a = 0

def func():
    global a
    a += 1

for i in range(5):
    func()

print(a)

람다 표현식

람다 표현식을 이용하면 함수를 매우 간단하게 작성하여 적용할 수 있다.

def add(a, b):
    return a + b

print(add(2, 8))

print((lambda a, b: a + b)(3, 7))