[프로그래머스/파이썬]코딩테스트 입문 - 점의 위치 구하기

점의 위치 구하기

사분면은 한 평면을 x축과 y축을 기준으로 나눈 네 부분입니다. 사분면은 아래와 같이 1부터 4까지 번호를매깁니다.

• x 좌표와 y 좌표가 모두 양수이면 제1사분면에 속합니다.
• x 좌표가 음수, y 좌표가 양수이면 제2사분면에 속합니다.
• x 좌표와 y 좌표가 모두 음수이면 제3사분면에 속합니다.
• x 좌표가 양수, y 좌표가 음수이면 제4사분면에 속합니다.

x 좌표 (x, y)를 차례대로 담은 정수 배열 dot이 매개변수로 주어집니다. 좌표 dot이 사분면 중 어디에 속하는지 1, 2, 3, 4 중 하나를 return 하도록 solution 함수를 완성해주세요.

def solution(dot):
    answer = 0
    if dot[0]*dot[1] > 0:
        if dot[0] > 0 : answer = 1
        else: answer = 3
    else:
        if (dot[0] > 0) :
            answer = 4
        else:
            answer = 2
    return answer

가장 먼저 생각난건 if 문으로 간단하게 분류를 하는 것이기 때문에, x좌표와 y좌표를 곱했을 때 0보다 크고 작음을 이용해 두개씩 나누어 주고 다시 조건문을 사용했다. 내가 만든 코드는 이쁜 코드는 아니다. 나랑 똑같은 코드를 이쁘게 바꾸면 아래와 같다.

def solution(dot):
    if dot[0]*dot[1] > 0:
        return 1 if dot[0] > 0 else 3
    else:
        return 4 if dot[0] > 0 else 2