Python Grammar - Input

 

잠깐 안 하면 까먹는 기초 문법을 정리해봐야겠다.

 

1. 한 줄에 2개 입력받기

a, b = map(int, input().split())

 

2. 한 줄에 여러개 입력받기

nums = [int(x) for x in input().split()]

 

3. n개의 줄에 하나씩 입력받기

nums = [int(input()) for _ in range(n)]

 

4. ?개의 줄에 하나씩 입력받기 (조건에 따라 끝남)

nums = []

while True:
    user_input = input()
    if user_input == "조건":
        break
    nums.append(int(user_input))

print(nums)

# 또는

# -1 이 입력됐을 때 끝내기
nums = [int(x) for x in iter(lambda: input(), "-1")]

 

 

5. n개의 줄에 여러 정수를 입력받아 리스트에 저장하기 

 

5-1. expend : [10, 20, 30, 40, 50, 60, 70, 80, 90]

# input : 
# 10 20 30
# 40 50
# 60 70 80 90

nums = []

for _ in range(n):
    user_input = input()
    nums.extend([int(x) for x in user_input.split()])

 

 

5-2. append : [[10, 20, 30], [40, 50], [60, 70, 80, 90]]

# input : 
# 10 20 30
# 40 50
# 60 70 80 90

리스트명 = [list(map(int, input().split())) for _ in range(n)]

 

 

6. n개에 줄에 걸쳐 문자열 입력받아 리스트로 저장하기

words = [input() for _ in range(n)]

 

7. n개의 줄에 n개 입력 받기

answer = [[int(x) for x in input().split()] for _ in range(n)]