[Python] 데이터 입력받기

💡 input()

1. 정수를 한줄에 입력받아 리스트에 저장

> data = input()
입력> 1 2 3 4

 

결과 : data = '1 2 3 4'

> data = input().split() 
입력> 1 2 3 4

 

결과 : data = ['1', '2', '3', '4']

split()은 받은 문자열을 특정문자를 기준(디폴트는 공백문자)로 나눠 리스트로 저장한다.

map()은 해당 리스트의 모든 원소에 int()를 적용한다.

> data = map(int, input().split()) 
> 1 2 3 4

 

결과 : data = <map object at 0x7ffafd980f40>

최종적으로 결과를 다시 list()로 바꿔준다.

 

⭐ 최종 코드⭐

> data = list(map(int, input().split())) 
> 1 2 3 4

 

결과 : data = [1, 2, 3, 4]

최종적으로 숫자 자료형으로 이루어진 리스트를 얻게된다!

 

2. 정수를 한줄로 입력받아 변수에 저장

맵 객체는 결과를 변수 여러 개에 저장하는 언패킹(unpacking)이 가능하다.

> a,b,c = map(int, input().split())
> 1 2 3 

결과 : a=1 , b=2, c=3

데이터가 적을때에는 이와 같이 사용한다.

---

💡 sys.stdin.readline()

입력의 개수가 많은 경우에는 input()을 사용하지 않는다. 파이썬의 기본 input()함수는 동작속도가 느려서 시간초과가 될 수 있기 때문이다. 이 경우 파이썬의 sys.stdin.readline()을 이용한다.

1. 한개의 정수 입력받기

a = int(sys.stdin.readline())

b = sys.stdin.readline().rstrip()

sys.stdin.readline()은 한줄 단위로 입력받기 때문에, 개행문자가 같이 입력 받아진다.
만약 1을 입력했다면, 1\n 이 저장되기 때문에, 개행문자를 제거해야한다.

단, a는 정수형이고 b는 문자열이다.

2. 여러 개의 정수를 한 줄에 입력받아 리스트에 저장⭐⭐⭐

input()대신에 sys.stdin.readline()을 써주면된다.

import sys
data = list(map(int, sys.stdin.readline().split()))

 

3. 정수를 n줄 입력받아 리스트에 저장

import sys
n = int(input("데이터의 수(엔터로 구분) : "))
data = [int(sys.stdin.readline()) for i in range(n)]

또는

import sys
n = int(input("데이터의 수(엔터로 구분) : "))
data = [int(input) for i in range(n)]

4. 정수를 n줄 입력받아 2차원 리스트에 저장

import sys

data = []
n = int(input("데이터의 수 : "))
for i in range(n):
	data.append(list(map(int,sys.stdin.readline().split())))

 

---

💡 관련 알고리즘 문제

대표적 예시는 백준 BOJ 15552번 문제이다.

반복문으로 여러줄 입력받는 상황에서는 반드시 sys.stdin.readline()을 사용해야 시간초과가 발생하지 않는다.

 

import sys 
n = int(input()) #갯수

for i in range(n): 
	a,b = map(int, sys.stdin.readline().split()) 
	print(a+b)

---

+ 데이터 출력하기

# 자동 줄바꿈
print(1,2)

# 줄바꿈을 원하지 않을때
print(3, end=" ")

# 숫자는 문자열로 바꾼 후에 사용
answer = 7
print("정답은 "+str(answer)+"입니다.")