python 두 수를 입력받아 합을 구하는 함수

두 자연수를 입력받아 그 중에 작은 값부터 큰 값 사이에 있는 모든 자연수를 더하는 함수입니다.

 

인자를 하나만 입력했다면 1부터 입력값까지의 합을 계산합니다.

 

def range_sum(a, b = None):
    sum = 0
    if (None == b):
        b = a
        a = 1
    for num in range(a, b+1):
        sum += num
    print(sum)

if __name__ == "__main__":
    range_sum(1)
    range_sum(2)
    range_sum(5)
    range_sum(1, 5)
    range_sum(4, 5)

 

[실행결과]

1
3
15
15
9