[이것이 취업을 위한 코딩 테스트다 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(=seperator) 를 사용하면 이를 기준으로 사이에 두고 값을 출력한다.

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