함수

문구점에서 물건 세 개를 10% 할인해 파는 프로그램을 만든다고 해 봅시다. 할인가는 가격에 0.9를 곱하고 반올림하면 됩니다. 함수 없이 쓰면 이렇게 됩니다.

print(round(12000 * 0.9))
print(round(8500 * 0.9))
print(round(23000 * 0.9))

10800, 7650, 20700이 나옵니다. 잘 실행됩니다. 그런데 같은 모양의 계산이 세 번 나와 있는 것이 눈에 걸립니다. 실제로 문제가 되는 것은 할인율이 바뀔 때입니다. 20% 할인으로 바뀌면 0.9를 세 군데에서 찾아 0.8로 고쳐야 합니다. 하나를 빠뜨리면 물건 하나만 값이 다르게 나오는데, 오류는 나지 않습니다.

같은 일에 이름을 붙여 한 군데에만 적어 두면 이 문제가 사라집니다.

def discount(price):
    return round(price * 0.9)


print(discount(12000))
print(discount(8500))
print(discount(23000))

이제 20% 할인으로 바뀌어도 고칠 곳은 한 군데뿐입니다. discount 안의 0.9를 0.8로 바꾸면 세 곳의 결과가 한꺼번에 바뀝니다.

자판기에 돈을 넣으면 음료가 나옵니다. 안에서 어떻게 작동하는지는 몰라도 됩니다. 무엇을 넣으면 무엇이 나오는지만 알면 쓸 수 있습니다. 함수도 그렇습니다.

매개변수는 하나가 아니어도 됩니다. 쉼표로 나누어 여러 개를 적습니다. 할인율까지 매개변수로 전달받으면 한 함수로 여러 할인율을 다룰 수 있습니다.

def discount(price, rate):
    return round(price * (1 - rate))


print(discount(12000, 0.1))
print(discount(12000, 0.25))

10800과 9000이 나옵니다. 순서가 중요합니다. 앞의 값이 price, 뒤의 값이 rate가 됩니다. discount(0.1, 12000)처럼 순서를 바꿔 호출하면 엉뚱한 값이 나옵니다.

함수 안에서 변수를 하나 만들고, 함수 밖에서 그 변수를 사용하면 어떻게 될까요?

def make_total(price):
    total = price * 3
    return total


print(make_total(1000))
print(total)

3000이 출력되고 나서 오류가 나며 멈춥니다. total이라는 이름이 정의되어 있지 않다는 것입니다. 분명히 함수 안에서 만들었는데 말입니다.

함수 안에서 쓴 이름이 밖에도 남는다면, 여러 함수가 우연히 같은 이름을 쓰는 순간 서로의 값을 덮어씁니다. total이나 i처럼 흔한 이름이라면 거의 반드시 부딪힙니다. 그러면 함수를 하나 가져다 쓸 때마다 그 안에서 무슨 이름을 쓰는지 전부 확인해야 합니다. 자판기 안을 열어 봐야 하는 셈입니다.

큰 문제를 만나면 손댈 수 있는 크기로 쪼개는 것이 첫걸음입니다. 「점수를 보고 등급을 알려 주는 프로그램」도 통째로 놓으면 막막하지만, 쪼개 보면 두 개의 작은 문제입니다.

두 조각 모두 「무엇을 전달받아 무엇을 반환하는가」가 뚜렷합니다. 그러면 그대로 함수 하나씩이 됩니다.

def average(values):
    return sum(values) / len(values)


def grade(value):
    if value >= 80:
        return 'A'
    elif value >= 60:
        return 'B'
    else:
        return 'C'


scores = [88, 74, 95, 61, 79]
result = average(scores)

print(round(result, 1), grade(result))

79.4 B가 나옵니다. 함수 정의 아래의 세 줄은 점수를 놓고, 평균을 구하고, 등급을 붙입니다. 코드가 문제를 쪼갠 순서 그대로 읽힙니다.

scores = [88, 74, 95, 61, 79]

print(len(scores))
print(max(scores))
print(min(scores))
print(round(sum(scores) / len(scores), 1))
print(sorted(scores))