코딩테스트 파이썬 문법 모음집 (1)

빠른 입출력

import sys
S = sys.stdin.readline().rstrip()
N = int(sys.stdin.readline().rstrip())

sys.stdout.write(S+"\n") #개행 문자가 없으므로 유의

input = sys.stdin.readline
print = sys.stdout.write
S = input().rstrip()
N = int(input().rstrip())

print(S+"\n")
print(str(N) + "\n")

filter 함수

nums=[3,5,4,8,9]
nums= list(filter(lambda n: n%2 != 0, nums))
print(nums) 
# [3, 5, 9]

List Comprehension 사용

nums = [3,5,4,8,9]
nums = [ n for n in nums if n%2 != 0]
print(nums)
# [3, 5, 9]

enumerate

for i, c in enumerate("abcd"):
	print(f'{i}. {c}')
# 0. a
# 1. b
---

문자열과 숫자

숫자 각 자리의 합

I = 245
ans = sum(map(int, str(I)))
print(ans)

입력이 문자인지 숫자인지 확인

S = input()
print(S.isdigit())

배열

배열 정렬 여부 확인

L == L.sorted()

배열 뒤집기

L = L[::-1]

배열 정렬

# sort() : 원본 배열 변경
L = [3, 2, 4, 5]
L.sort()
print(L) # [2, 3, 4, 5]

L = [3, 2, 4, 5]
L.sort(reverse=True)
print(L) # [5, 4, 3, 2]

# sorted() : 원본 배열 변경 X
L = [3, 2, 4, 5]
L = sorted(L)
print(L) # [2, 3, 4, 5]

L = [3, 2, 4, 5]
L = sorted(L, reverse=True)
print(L) # [5, 4, 3, 2]

배열 -> 문자열 변환

: 구분자.join(리스트)

L = [5, 3, 1, 7]
print(" ", join(L)) # '5 3 1 7'
print(",", join(L)) # '5,3,1,7'

배열에서 특정한 원소들만 제거

L = [1, 2, 3, 4, 5, 4, 3, 2, 1]
remove_set =  {2, 4}
L = [n for n in L if n not in remove_set]
print(L)
# [1, 3, 5, 3, 1]

배열 원소의 합

• sum()

L = [2,4,6]
print(sum(L))  # 12

• zip()

A = [1, 2, 3, 4]
B = ['a', 'b', 'c', 'd']
print(list(zip(A, B))) # [(1, 'a'), (2, 'b'), (3, 'c'), (4, 'd')]

딕셔너리

key로 value 조회

D = dict({5: 1, 3: 3, 4: 2})
print(D.get(5)) #1
print(D.get(999, 0)) #0 ==> 찾으려는 값이 없을 때는 2번째 값 출력