[Python] 데이터 입출력 방법

JavaScript를 사용해 백준, 프로그래머스와 같은 다양한 플랫폼에서 알고리즘 문제를 풀어왔다. 

최근에는 인공지능의 한 분야인 컴퓨터 비전에 관심을 가지면서, 파이썬에 익숙해지기 위해 언어를 전환하기로 했다. 

그 첫걸음으로, 파이썬에서 가장 기본이 되는 데이터 입출력 방법을 알아보고 정리하는 시간을 가지려고 한다.

 

한 줄의 문자열을 입력 받기

input() 함수는 한 줄의 문자열을 입력받는 기능을 제공한다. 입력받은 데이터를 정수형으로 처리하려면, int() 함수를 사용해 문자열을 정수로 변환해야 한다.

# 데이터의 개수 입력
n = int(input())

 

여러 개의 데이터를 입력 받기

여러 개의 데이터를 입력받을 때는 보통 공백으로 구분된 데이터를 처리하게 된다. 이 경우 list(map(int, input().split()))을 사용하면, 입력된 문자열을 공백 단위로 나누어 정수형 리스트로 변환할 수 있다.

공백으로 구분된 데이터의 개수가 많지 않다면, 단순히 map(int, input().split())을 이용하는 것도 가능하다.

# 각 데이터를 공백으로 구분하여 입력
data = list(map(int, input().split()))

# n, m, k를 공백으로 구분하여 입력
n, m, k = map(int, input().split())

 

여러 개의 데이터를 입력 받기(입력의 개수가 많은 경우)

입력의 개수가 많을 경우, 기본 input() 함수는 동작 속도가 느려 시간 초과로 오답 판정을 받을 수 있다. 이럴 때는 sys 라이브러리의 sys.stdin.readline() 함수를 사용하면 더 빠르게 입력을 처리할 수 있다.

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

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

 

여러 개의 데이터를 입력 받기(입력의 개수가 정해져 있지 않은 경우)

입력의 개수가 정해져 있지 않다면, 일반적으로 입력이 끝날 때까지 반복해서 데이터를 읽어야 한다. 파일 입력 또는 여러 줄 입력에서 많이 사용하는 방식으로 try-except를 사용한다.

while True:
    try:
        a, b = map(int, input().split())
    except EOFError:
        break

EOF(End Of File)Error는 입력이 더 이상 들어오지 않을 때 발생하는 예외이므로 이를 통해 입력을 종료할 수 있다.

 

출력하기

print() 함수는 여러 변수를 콤마(,)로 구분해 매개변수로 전달할 수 있으며, 이 경우 변수들은 띄어쓰기로 구분되어 출력된다. 또한, print()는 기본적으로 출력 후 자동으로 줄 바꿈을 수행한다.

# 출력할 변수들
a = 1
b = 2

# 변수 출력
print(a, b) # 1 2

# 출력 줄 바꿈
print(a) # 1
print(b) # 2

 

일부 문제에서는 문자열과 숫자를 함께 출력해야 하는 경우가 있다. 이때 단순히 + 연산자를 사용해 문자열과 숫자를 더하면 오류가 발생하므로, str() 함수를 사용해 숫자를 문자열로 변환해 해결할 수 있다. 또한, f-string 문법을 사용하면 {} 안에 변수를 넣는 방식으로 자료형 변환 없이도 문자열과 숫자를 간단히 함께 출력할 수 있다.

# 출력할 변수들
answer = 7

print("정답은 " + str(answer) + "입니다.")
print(f"정답은 {answer}입니다.")

 

* 본 내용은 <이것이 취업을 위한 코딩 테스트다 with 파이썬(나동빈)>을 참고하여 작성하였습니다.