헷갈리는 python 문법 -- input()

1. 숫자 여러 개 입력 받기 

10 20 30 40

nums = list(map(int, input().split()))
print(nums)

[10, 20, 30, 40]

 

• input() → '10 20 30 40' (문자열)
• split() → ['10', '20', '30', '40'] (문자열 리스트)
• map(int, ...) → [10, 20, 30, 40] (정수 리스트)
• list()는 map()이 반환하는 이터레이터를 리스트로 바꿔줌 < map()은 이터레이터 객체이기 때문에 list()가 필요한 것

---

2. 공백 없는 숫자 여러 개 입력 받기 

101111

nums = list(map(int, input().strip()))
print(nums)

[1, 0, 1, 1, 1, 1]

 

---

3. 문자 여러 개 입력받기

a b c d

chars = input().split()
print(chars)

['a', 'b', 'c', 'd']

• 문자열은 따로 int()로 바꿀 필요 없이 split()만 써도 바로 리스트로 사용 가능
• 그래서 list(map(...)) 필요 없음

 

+ 아래처럼 코드를 짰다가 백준 답이 여러번 틀리다고 나왔음 ㅠ

chars = list(input().split(' '))
print(chars)

['a', 'b', 'c', 'd\']

---

4. 공백 없는 문자열을 문자 하나씩 쪼개고 싶을 때

<실수하기 쉬운 코드>

ABCD

s = input().split()
print(s)

['ABCD']

 

<올바른 코드>

s = list(input())
print(s)

['A', 'B', 'C', 'D']

 

• split()은 공백 기준으로 나누는 함수이므로, 공백이 없으면 전체 문자열을 하나로 반환
• 글자 단위로 쪼개고 싶다면 list()를 사용해야 함