[코딩테스트 대비] python 메서드 및 라이브러리 활용 정리

메서드 

 

- 전체 출력 결과를 한줄에 출력하기

for i in n:
    print(ord(i)-64,end=' ') # 1 2 3 4 ...25 26 으로 출력

- 입력값이 서로 붙어있을 땐, input()/split()이 아닌, 그냥 input()만 써서 작성

#입력값
#12345
#67890
graph = [list(map(int,input())) for _ in range(2)]
#print(graph) = [[1, 2, 3, 4, 5], [6, 7, 8, 9, 0]]

- 절대값 출력

print(abs(-5)) #5 출력

라이브러리

내장함수

- eval() 함수 : 수학 수식이 문자열 형식으로 들어오면 수식의 계산 결과를 반환한다.

result = eval((3+5)*7)
print(result) #56

- sorted()함수 : 오름차순으로 정리하여 결과를 반환한다.

#오름차순
result = sorted([9,1,8,5,4])
print(result) # [1,4,5,8,9]

#내림차순
result = sorted([9,1,8,5,4], reverse = True)
print(result) # [9,8,5,4,1]

math 라이브러리

- factorial() 함수

import math
print(math.factorial(5)) #5! => 120 출력

- sqrt() 함수

import math
print(math.sqrt(7)) # 7의 제곱근 반환 => 2.6457....

(응용) sqrt를 활용하여 O(n**(1/2))수준으로 특정 숫자가 소수인지 판별하는 알고리즘을 짤 수 있습니다.

import math

def is_prime_number(x):
	for i in range(2,int(math.sqrt(x))+1):
		if x % i == 0:
			return False
	return True

print(is_prime_number(4)) #False . 4는 소수가 아니다!!

(itertools,heapq,bisect은 추후 다룰 예정입니다!) permutation도,,

 

SWEA에서 사용한 적이 없... 읍읍

 

또 문제 풀면서 알아두면 좋을 것 같은 메서드를 발견하면 수시로 업데이트하겠습니다.

 

감사합니다!