코딩테스트를 위한 Python 입력 처리

다시 한번 정리해보는 Python 입력 처리 

 

input() 대신 sys.stdin.readline()으로 입력 받기

import sys 

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

 

한 개의 정수를 입력 받기

import sys

n = int(sys.stdin.readline())
print(n)

 

정해진 개수의 정수를 한줄에 입력받기

import sys

a, b, c = map(int, sys.stdin.readline().split())
# map()을 이용해 a, b ,c 각각을 int형으로 형변환 

print(a, b, c)

 

임의의 개수의 정수를 한줄에 입력받아 리스트에 저장하기

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

 

임의의 개수의 정수를 n줄 입력받아 2차원 리스트에 저장하기

import sys

data = []
n = int(sys.stdin.readline())
for i in range(n):
    data.append(list(map(int, sys.stdin.readline().split())))

input

3
1 2 3
4 5 6
7 8 9

output

[[1, 2, 3], [4, 5, 6], [7, 8, 9]]

 

문자열 n줄 입력받아 리스트에 저장하기

import sys

n = int(sys.stdin.readline())
data = [sys.stdin.readline().strip() for i in range(n)]

print(data)

input

3
apple
banana
carrot

output

['apple', 'banana', 'carrot']

 

이어진 숫자를 한자리씩 나눠 리스트로 저장

# 문자열 리스트 
arr = list(input())
print(arr)

# 숫자형 리스트
arr = list(map(int, input()))
print(arr)

 

N행으로 이뤄진 2차원 배열 입력받기

조건 : 배열의 행 크기(N)를 알아야 한다. 

N = int(input())

# 1. 원소를 하나씩 입력받기 
arr = [for _ in range(N)] // 2차원 배열의 가로길이 : B
for i in range(N):    
	arr[i] = list(map(int, input().split()))


# 2. 원소에 list 추가하기
arr = []

for i in range(B):    
	arr.append(list(map(int, input().split())))


# 3. 선언과 동시에 입력받기 
arr = [list(map(int, input().split())) for _ in range(N)]

 

사용자 입력으로 딕셔러니 생성하기

dict_list = []
n = int(input())
for i in range(n):
    my_dict = {} # empty dictionary
    key = input('키 입력 : ')
    val = input('밸류 입력 : ')
    my_dict[key] = val
    dict_list.append(my_dict)
print(dict_list)

input

3
키 입력 : key1
밸류 입력 : val1
키 입력 : key2
밸류 입력 : val2
키 입력 : key3
밸류 입력 : val3

output

[{'key1': 'val1'}, {'key2': 'val2'}, {'key3': 'val3'}]