[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()을 사용해야 한다.

 

출력하기

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 파이썬(나동빈)>을 참고하여 작성하였습니다.