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

기본 입출력

• 모든 프로그램은 적절한 (약속된) 입출력 양식을 가지고 있다.
• 프로그램 동작의 첫 번째 단계는 데이터를 입력 받거나 생성하는 것이다.
• ex) 학생의 성적 데이터가 주어지고, 이를 내림차순으로 정렬한 결과를 출력하는 프로그램

자주 사용되는 표준 입력 방법

• input() 함수는 한 줄의 문자열을 입력 받는 함수이다.
• map() 함수는 리스트의 모든 원소에 각각 특정한 함수를 적용할 때 사용한다.
• ex) 공백을 기준으로 구분된 데이터를 입력받을 때는 다음과 같이 사용한다.
  • list(map(int, input().split()))
• ex) 공백을 기준으로 구분된 데이터의 개수가 많지 않다면, 단순히 다음과 같이 사용한다.
  • a, b, c = map(int, input().split())

입력을 위한 전형적인 소스코드 1)

# 데이터의 개수 입력
n = int(input()) # 5 (입력)
# 각 데이터를 공백을 기준으로 구분하여 입력
data = list(map(int, input().split())) # 65 90 75 34 99 (입력)

data.sort(reverse=True)
print(data) # [99, 90, 75, 65, 34]

입력을 위한 전형적인 소스코드 2)

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

print(n, m, k) # 3 5 7

빠르게 입력 받기

• 사용자로부터 입력을 최대한 빠르게 받아야 하는 경우가 있다.
• 파이썬의 경우 sys 라이브러리에 정의되어 있는 sys.stdin.readline() 메서드를 이용한다.
  • 단, 입력 후 엔터(Enter)가 줄 바꿈 기호로 입력되므로 rstrip() 메서드를 함께 사용한다.
  • 이진탐색, 정렬, 그래프 문제에서 자주 사용된다.

import sys

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

자주 사용되는 표준 출력 방법

• 파이썬에서 기본 출력은 print() 함수를 이용한다.
  • 각 변수를 콤마(,)를 이용하여 띄어쓰기로 구분하여 출력할 수 있다.
• print() 는 기본적으로 출력 이후에 줄 바꿈을 수행한다.
  • 줄 바꿈을 원치 않는 경우 'end' 속성을 이용할 수 있다.

출력을 위한 전형적인 소스코드

# 출력할 변수들
a = 1
b = 2
print(a, b) # 1 2
print(7, end=" ")
print(8, end=" ")

# 출력할 변수
answer = 7
print("정답은 " + str(answer) + "입니다.")
# 7 8 정답은 7입니다.

f-string 예제

• 파이썬 3.6부터 사용 가능하며, 문자열 앞에 접두사 'f'를 붙여 사용한다.
• 중괄호 안에 변수명을 기입하여 간단히 문자열과 정수를 함께 넣을 수 있다.

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

---

'이것이 취업을 위한 코딩 테스트다 with 파이썬'의 교재를 보고 작성한 내용입니다.