[이것이 취업을 위한 코딩 테스트다 with 파이썬] Chapter 06. 입출력

1. 데이터 입력받기 

파이썬에서 입력받을 때는 input()을 이용함. 이는 한 줄의 문자열을 입력받도록 해준다.

여러 개의 데이터를 입력받을 때는 데이터가 공백으로 구분되는 경우가 많으므로, 아래와 같이 입력받은 문자열을 띄어쓰기로 구분하여 각각 정수 자료형의 데이터로 저장하는 코드의 사용 빈도가 매우 높음.

 

 list(map(int, input().split())) 

input()으로 입력받은 문자열을 split()을 통해 공백으로 나눈 리스트로 바꾼 후, map을 이용하여 해당 리스트의 모든 원소에 int() 함수를 적용한다.

# 데이터의 개수 입력
n = int(input()) # 입력값 : 5
# 각 데이터를 공백으로 구분하여 입력
data = list(map(int, input().split())) # 입력값 : 1, 2, 3, 4, 5

# 공백을 기준으로 구분하여 데이터 입력
n, m, k = map(int, input().split())

 

입력의 개수가 많아지는 경우에 input() 함수를 그대로 사용하게 되면 시간 초과가 날 수 있다는 단점이 있다. 이때 사용하는 함수가 sys 라이브러리 내의 sys.stdin.readline() 함수이다. input() 함수와 마찬가지로 한 줄씩 입력받기 위해 사용된다.

sys 라이브러리 사용 시, 한 줄 입력을 받고 나서 꼭 rstrip() 함수를 호출해주어야 한다.

Why? → readline()으로 입력하면 입력 후 Enter 가 줄 바꿈 기호로 입력되므로, 이 공백 문자를 제거하기 위해 호출해주어야 한다.

import sys

# 문자열 입력받기
sys.stdin.readline().rstrip()

 

 

2. 데이터 출력하기 

파이썬에서 출력할 때는 print()를 사용한다. 이는, 각 변수를 콤마, 로 구분하여 매개변수로 넣을 수 있다.

또한, print()는 기본적으로 출력 이후 줄 바꿈을 수행한다.

a = 1
b = 2
print(a, b) # 출력값 : 1 2

a = 1
b = 2
print(a) # 출력값 1
print(b) # 출력값 2

 

문자열 자료형과 함께 정수형 자료를 출력하고자 할 때 str() 함수를 이용한다.

answer = 7
print("정답은 " + str(answer) + "입니다.") # 출력문 : 정답은 7입니다.

 

만약, 자료형의 변환 없이도 간단하게 문자열과 정수형 자료를 함께 출력하고자 할 경우, f-string 문법을 사용하면 된다.

문자열 앞에 접두사 f를 붙이고, 중괄호 {} 안에 변수를 넣으면 된다.

answer = 7
print(f"정답은 {answer}입니다.") # 출력문 : 정답은 7입니다.

 

split을 사용하면 이를 기준으로 문자열 자른다. 또한, sep(=separator)를 사용하면 이를 기준으로 사이에 두고 값을 출력한다.

a, b = input().split(':') # 입력문 : 3:16
print(a, b, sep=':') # 출력문 : 3:16