[코딩 테스트를 위한 파이썬 문법] 입출력

본 포스팅은 나동빈 저자의 ‘이것이 취업을 위한 코딩테스트’를 공부하며 정리한 노트입니다.

input()

데이터를 입력받을 때는 input()을 이용한다. 한 줄의 문자열을 입력받도록 해준다.

여러 개의 데이터를 입력 받을 때는 데이터가 공백으로 구분되는 경우가 많다. 이때는, list(map(int, input().split()))을 이용하면 된다.

많은 문제는 공백, 혹은 줄 바꿈을 기준으로 데이터를 구분한다. 파이썬에서는 구분자가 줄 바꿈인지 공백인지에 따라서 다른 처리를 요구한다.

줄 바꿈: int(input())

공백: list(map(int, input().split())

공백으로 구분된 데이터가 많지 않다면, map(int, input(). split())을 이용한다.

sys.stdin.readline()

입력을 최대한 빠르게 받아야 하는 경우가 있다. 파이썬 sys 라이브러리에 정의되어 있는 sys.stdin.readline() 함수를 이용한다.

import sys
data = sys.stdin.readline().rstrip()

sys 라이브러리를 사용할 때는 한 줄 입력을 받고 나서 rstrip() 함수를 꼭 호출해야 한다. readline()으로 입력하면 입력 후 엔터가 줄 바꿈 기호로 입력되는데, 이 공백 문자를 제거하려면 rstrip()을 사용해야 한다.

print()

파이썬의 print()는 출력 이후에 줄 바꿈을 수행한다.

문자열과 수를 함께 출력해야 되는 경우에는 3가지 방법으로 해결한다.

answer = 7

# str() 함수를 이용하여 출력하고자 하는 변수 데이터를 문자열 바꾼다.
print("정답은 " + str(answer) + "입니다.")

# 각 변수를 콤마(,)로 구분하여 출력한다.
print("정답은", str(answer), "입니다.")

# f-string 문법
print(f"정답은 {answer}입니다.")