[python] 코딩테스트를 위한 파이썬 문법 정리

1. deque 

- 선입선출 큐

from collections import deque

deq = deque()

deq.append(1)
deq.appendleft(2)
deq.pop()
deq.popleft()

• deque.append(item): item을 데크의 오른쪽 끝에 삽입
• deque.appendleft(item): item을 데크의 왼쪽 끝에 삽입
• deque.pop(): 데크의 오른쪽 끝 엘리먼트를 가져오는 동시에 데크에서 삭제한다.
• deque.popleft(): 데크의 왼쪽 끝 엘리먼트를 가져오는 동시에 데크에서 삭제한다.
• deque.remove(item): item을 데크에서 찾아 삭제한다.
• deque.rotate(num): 데크를 num만큼 회전한다(양수면 오른쪽, 음수면 왼쪽).

 

2. Counter

from collections import Counter

a =['red', 'blue', 'red', 'green', 'blue', 'blue']
b = [1,2,2,3,3,3,4,4,4,4]

print(Counter(a))			# Counter({'blue': 3, 'red': 2, 'green': 1})
print(Counter(a)['red'])		# 2

print(Counter(b))			# Counter({4: 4, 3: 3, 2: 2, 1: 1})
print(Counter(b)[3])			# 3

• Counter(item) : list 또는 set인 입력을 넣으면 각 항목을 key로 개수를 value로 리턴해준다
• Counter.most_common(item) : item 수 만큼의 상위 요소만 출력
• Counter 끼리 덧셈, 뺄셈 가능
• &(교집합), |(합집합) 연산 가능

 

3. heapq

- 데이터를 정렬된 상태로 저장

import heapq

heap = []
heapq.heappush(heap, 4)
heapq.heappush(heap, 1)
heapq.heappush(heap, 7)
heapq.heappush(heap, 3)
print(heap)				# [1, 3, 7, 4]

print(heapq.heappop(heap))		# 1
print(heapq.heappop(heap))		# 3
print(heapq.heappop(heap))		# 4
print(heapq.heappop(heap))		# 7

• heapq.heappush(heap, item) : heap 에 item(원소) 추가
• heapq.heappop(heap) : heap 의 가장 작은 원소를 삭제 후 리턴
• heapq.heapify(list) : 이미 원소가 들어있는 리스트를 힙으로 반환 

 

4. permutations, combinations 

- 순열과 조합

from itertools import permutations
from itertools import combinations


arr = ['A', 'B', 'C']
nPr = permutations(arr, 2)
nCr = combinations(arr, 2)

print(list(nPr))		# [('A', 'B'), ('A', 'C'), ('B', 'A'), ('B', 'C'), ('C', 'A'), ('C', 'B')]
print(list(nCr))		# [('A', 'B'), ('A', 'C'), ('B', 'C')]

• permutations(arr,x) : 리스트 arr에서 x개를 골라 순열을 반환 
• combinations(arr,x) : 리스트 arr에서 x개를 골라 조합을 반환

 

5. eval()

a = eval("1+3*5")
b = eval("{'red':3,'blue':4}")
c = eval("[1,2,3]")

print(a,type(a))		# 16 <class 'int'>
print(b,type(b))		# {'red': 3, 'blue': 4} <class 'dict'>
print(c,type(c))		# [1, 2, 3] <class 'list'>

 

6. re.split()

import re

a = '12+3*4-88'
x = re.split('[+*-]',a)
y = re.split('([+*-])',a)

print(x)	# ['12', '3', '4', '88']
print(y)	# ['12', '+', '3', '*', '4', '-', '88']

 

7. ord(), chr()

print(ord('a'))		# 97
print(chr(97))		# a

• ord() : 문자 -> 아스키코드
• chr() : 아스키코드 -> 문자

 

8. 이차원 리스트 정렬

arr = [[2, 1], [3, 4], [1, 2], [1, 3], [3, 2]]

a = sorted(arr, key=lambda n : n[0])
b = sorted(arr, key=lambda n : n[1])

c = sorted(arr, key=lambda n : (n[0],n[1]))
d = sorted(arr, key=lambda n : (n[0],-n[1]))
e = sorted(arr, key=lambda n : (n[1],n[0]))

print(a)		# [[1, 2], [1, 3], [2, 1], [3, 4], [3, 2]]
print(b)		# [[2, 1], [1, 2], [3, 2], [1, 3], [3, 4]]

print(c)		# [[1, 2], [1, 3], [2, 1], [3, 2], [3, 4]]
print(d)		# [[1, 3], [1, 2], [2, 1], [3, 4], [3, 2]]
print(e)		# [[2, 1], [1, 2], [3, 2], [1, 3], [3, 4]]

lambda 를 쓰면 arr 이 순회하면서 n 에 하나씩 들어간다고 이해하면 됨 

• sorted(arr, key=lambda n : n[0]) : arr의 첫 번째 값으로 오름차순 정렬
• sorted(arr, key=lambda n : n[1]) : arr의 두 번째 값으로 오름차순 정렬
• sorted(arr, key=lambda n : (n[0],n[1])) : arr의 첫 번째 값으로 오름차순 정렬 후, 두번 째 값으로 오름차순 정렬
• sorted(arr, key=lambda n : (n[0],-n[1])) : arr의 첫 번째 값으로 오름차순 정렬 후, 두번 째 값은 내림차순으로 정렬
• sorted(arr, key=lambda n : (n[1],n[0])) : arr의 두 번째 값으로 오름차순 정렬 후, 첫번 째 값으로 오름차순 정렬

 

9. 딕셔너리 정렬

sorted(iterable,key) 이기 때문에, 딕셔너리를 iterable 로 변경하기 위해 items() 메소드를 사용

dic = {
    2: 1,
    3: 4,
    5: 2,
    1: 3,
    4: 1
}
print(dic)		# {2: 1, 3: 4, 5: 2, 1: 3, 4: 1}
print(dic.items())	# dict_items([(2, 1), (3, 4), (5, 2), (1, 3), (4, 1)])

a = sorted(dic.items(), key=lambda n:n[0])	# key 값으로 정렬
b = sorted(dic.items(), key=lambda n:n[1])	# value 값으로 정렬

print(a)		# [(1, 3), (2, 1), (3, 4), (4, 1), (5, 2)]
print(b)		# [(2, 1), (4, 1), (5, 2), (1, 3), (3, 4)]

 

10. 리스트, 문자열 자주 사용하는 메서드

str ='abcDEc'
lst = [3,6,3,8,10]

print(str.find('c'))		# 2
print(str.find('f'))		# -1

print(str.index('c'))		# 2
print(lst.index(3))		# 0

print(str[::-1])		# cEDcba
print(lst[::-1])		# [10, 8, 3, 6, 3]

print(str.upper())		# ABCDEC
print(str.lower())		# abcdec

print(str.replace('c','f'))	# abfDEf


print(list(set(str)))					# ['E', 'a', 'b', 'D', 'c']
print(''.join(list(set(str))))				# EabDc
print(sorted(''.join(list(set(str)))))			# ['D', 'E', 'a', 'b', 'c']
print(''.join(sorted(''.join(list(set(str)))))) 	# DEabc

print(list(set(lst)))					# [8, 10, 3, 6]
print(sorted(list(set(lst))))				# [3, 6, 8, 10]

 

▶ 문자열

• str.find(item) : 문자열에 x가 몇 번째 index에 처음 있는지 반환 없으면 -1 반환\
• str.upper(), 문자열.lower() : 문자열 대문자화, 소문자화
• str.replace(x,y) : 문자열에서 모든 x를 찾아서 y로 변경

▶ 리스트

• ''.join(list) :  리스트 내부에 있는 문자열들을 모두 연결

▶ 문자열 or 리스트

• x.index(item) : 문자열 또는 리스트에 item이 몇 번째 인덱스에 처음 있는지 반환
• x[::-1] : 문자열 또는 리스트 거꾸로 출력 
• list(set(x)) : 문자열 또는 리스트에서 변경되는 값 없애고 리스트로 저장

     * [::-1] : 리스트에서 하면 [1,2,3,4] -> [4,3,2,1] 되는데, [1234] -> [4321] 해주기 위해서는 문자열로 바꿔서 거꾸로 바꿔주고 다시 정수로 바꿔주면 된다

 

11. set

set1 = {1,2,3,4}
set2 = {3,4,5,6}
set3 = {3,4,5,6,7,8}

print(set1&set2)			# 교집합 {3, 4}
print(set1-set2)			# 차집합 {1, 2}
print(set1|set2)			# 합집합 {1, 2, 3, 4, 5, 6}
print(set1.issubset(set2))		# False
print(set2.issubset(set3))		# True

• set1 & set2 : set1과 set2의 교집합 반환
• set1 - set2 : set1에서 set2를 뺀 차집합 반환
• set1 | set2 : set1과 set2의 합집합 반환
• set1.issubset(set2) : set1이 set2의 부분 집합인지 반환

 

12. 문자열 검색

s = 'this is string'

print('is' in s)		# True
print(s.find('str'))		# 8
print(s.startswith('this'))	# True
print(s.endswith('string'))	# True

• 'a' in str : str에 'a'가 있는지
• str.startswith(item) : 문자열이 item으로 시작하는지 아닌지
• str.endswith(item) : 문자열이 item으로 끝나는지 아닌지

 

os

os.environ['HOME'] # 'HOME' 대신에 다양한 시스템 정보들이 들어갈 수 있습니다.
os.environ.get('HOME')
os.path.isfile('filename')
os.path.isdir('dirname')

# path가 존재하는 지 확인
os.path.exists(path) 

# 현재 경로에 folder 생성
os.mkdir(folder_name) 

# 지정된 path에 folder 생성하며 path에 포함된 branch 까지 생성
# 예를 들어 path가 ./A/B/C 이고 B/C가 없다면 마지막 C를 생성할 때까지 중간에 필요한 폴더 계층을 모두 생성합니다.
# mkdir과 유사하지만 중간에 sub directory 까지 한번에 만들어 줍니다.
os.makedirs(path) 

# 현재 디렉토리의 절대 경로를 출력함
os.path.abspath(os.getcwd())
# 경로 변경
os.chdir(path)

# cmd 명령어 입력
command = "ls -al"
os.system(command)

# 작업하고 있는 디렉토리 변경
os.chdir(path)

# 현재 프로세스의 작업 디렉토리 얻기
os.getcwd()

# 파일이나 디렉토리 지우기
os.remove(file_path or path)

# 파일의 상대 경로를 절대 경로로 바꾸는 함수
os.path.abspath(file_path)

# 주어진 경로의 파일이 있는지 확인하는 함수
os.path.exists(file_path)

# 현재 디렉토리 얻기
os.curdir()

# 부모 디렉토리 얻기
os.pardir()

# 디렉토리 분리 문자 얻기. . windows는 \ linux는 / 를 반환합니다.
os.sep() 

# 파일명만 추출
os.path.basename(file_path)

# 디렉토리 경로 추출
os.path.dirname(file_path)

# 마지막 디렉토리 폴더명 추출
path = "folder_A/folder_B/folder_C/folder_D"
os.path.basename(os.path.normpath(path))
# folder_D

# 경로와 파일명을 분리
os.path.split(file_path)

# 드라이브명과 나머지 분리 (MS Windows의 경우)
os.path.splitdrive(file_path)

# 확장자와 나머지 분리
os.path.splitext(file_path)

# 파일 사이즈 계산
MB = 1024 ** 2
GB = 1024 ** 3
# Mega Byte 단위로 표현
os.path.getsize(file_path) // MB
# Giga Byte 단위로 표현
os.path.getsize(file_path) // GB

import glob

path = "./*/*.py"
file_list = glob.glob(path)
file_list_py = [file for file in file_list]
# file_list_py = [file for file in file_list if file.endswith(".py")]

print ("file_list_py: {}".format(file_list_py))