[백준] 두 수 비교하기 (Python)

문제

두 정수 A와 B가 주어졌을 때, A와 B를 비교하는 프로그램을 작성하시오.

입력

첫째 줄에 A와 B가 주어진다. A와 B는 공백 한 칸으로 구분되어져 있다.

출력

첫째 줄에 다음 세 가지 중 하나를 출력한다.

• A가 B보다 큰 경우에는 '>'를 출력한다.
• A가 B보다 작은 경우에는 '<'를 출력한다.
• A와 B가 같은 경우에는 '=='를 출력한다.

제한

-10,000 ≤ A, B ≤ 10,000

예제 입력 1

1 2

예제 출력 1

<

예제 입력 2

10 2

예제 출력 2

>

예제 입력 3

5 5

예제 출력 3

==

첫 코드

A, B = map(int, input().split())

if A>B:
    return ">"
elif A<B:
    return "<"
else:
    return "=="

A, B를 입력하였을 때 input에 예제 입력과 같이 공백을 기준으로 2개의 숫자를 적을테지만 무조건 str으로 출력되므로, 공백을 기준으로 split()으로 나눠준 후 int를 부여하여 map시키면 A, B에 넣는 값이 숫자로 변환되게 된다
그치만 js에 익숙한 나머지 print()로 출력할 생각이 아닌 return을 적은 모습 ㅠㅠ

수정한 코드

# A B 입력 받았을 때 정수로 만드는 로직
A, B = map(int, input().split())

if A>B:
    print(">")
elif A<B:
    print("<")
else:
    print("==")