파이썬 빠른 입출력

map()

리스트의 모든 원소에 각각 특정한 함수를 적용할 때 사용한다!

map은 리스트의 요소를 지정된 함수로 처리해주는 함수이다

(map은 원본 리스트를 변경하지 않고 새 객체를 생성!)

map 함수 사용 안할 때

a = [1.2, 2.5, 3.7, 4.6]
for i in range(len(a)):
	a[i] = int(a[i])

map 함수 사용 시

a = [1.2, 2.5, 3.7, 4.6]
a = list(map(int, a))

지금까지 리스트에 대해서 map을 사용했는데, map은 모든 iterable한 객체를 넣을 수 있다!

여러 개의 변수를 담아내기

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

sys 모듈

위 방법보다 더 빠른 방법은 sys 모듈의 sys.stdin.readline()을 사용하는 것이다.

import sys

# 공백으로 구분된 2개의 숫자 입력 받기!
n, m = map(int, sys.stdin.readline().split())

# 2차원 리스트 생성
matrix = [list(map(int, sys.stdin.readline().split())) for _ in range(n)]

# 문자열 입력 받기
# 하나 입력시에 엔터가 사용되므로 rstrip()사용
data = sys.stdin.readline().rstrip()

strip()의 종류

• strip() : 인자로 전달된 문자를 string의 왼쪽과 오른쪽에서 제거한다.
• lstrip() : 인자로 전달된 문자를 string의 왼쪽에서 제거한다.
• rstrip() : 인자로 전달된 문자를 string의 오른쪽에서 제거한다.

2021-09-15