[프로그래머스] 다음에 올 숫자 (Python)

<내 풀이>

def solution(common):
    answer = 0
    if (common[1] - common[0] == common[2] - common[1]):
        answer = common[-1] + (common[1] - common[0])
    else:
        answer = common[-1] * (common[1] / common[0])
    return answer

리스트의 첫번째 요소와 두번째 요소의 차가 두번째 요소와 세번째 요소의 차와 같다면 등차수열, 아닐경우 등비수열
등차수열일 경우 마지막 값에 두번째 값과 첫번째 값의 차를 더함
등비수열일 경우 마지막 값에 두번째 값과 첫번째 값을 나눈 값을 곱합