[Python]각종 함수

리스트에 대해 인덱스와 값 쌍 출력하기 위한 방법

• enumerate()

반복문에 enumerate(리스트명) 사용하면 인덱스와 리스트 값 쌍이 나옴

# 예시

for i, key in enumerate(words):
	print(i, ': ', key)

 

문자열로 이뤄진 리스트나 딕셔너리에서 특정 문자로 시작하는지 확인하기 위한 함수 

• startswith()

key.startswith(num)

 

리스트에서 특정 원소의 인덱스를 반환해주는 함수 

• index()

key.index(num)==0

 

인접 리스트

# defaultdict 클래스로 그래프를 인접 리스트로 변환 할 수 있다. 
# defaultdict 클래스는 키가 없을 때 기본값을 defaultdict 형태로 기본값을 지정한다. 

# defaultdict 자료형에서는 append()를 통해 값을 추가할 수 있다. 

• defaultdict()
• append() 

from collections import defaultdict

tmp_list = defaultdict(list)

예를들어서, defaultdict(list)라고 코드 작성시 기본값은 []와 같이 초기화 한다.

tmp_list[start].append(end)

 

 

집합

• set() 집합 생성함수
• add() 집합에 객체 추가 함수 
• get() 키값을 통해 value를 조회하는 함수 

 

visited = set()

visited.add(node)

for injeop in node_tree.get(node, []):
# node_tree의 node에 대해 값이 있으면 해당 값의 개수만큼 개수를 출력해준다. injeop으로.
# 만약 값이 없으면, 빈 [] 리스트를 출력해준다.

 

 

우선순위 큐(Heap Queue)

 

파이썬 내장 모듈인  'heapq'는 최소 힙(min heap)자료구조를 제공한다. 

최소 힙은 가장 작은 요소가 항상 루트에 위치해, 최소값을 빠르게 찾을 수 있는 자료구조이다. 

 

• heappush() 힙큐에 요소를 추가하는 함수
• heappop() 힙큐에서 가장 작은 요소(루트에 위치한 요소)를 제거하고 반환하는 함

 

import heapq

# 최소 힙 생성
queue = []
heapq.heappush(queue, 5)
heapq.heappush(queue, 3)
heapq.heappush(queue, 8)

# 가장 작은 요소 제거 및 반환
smallest = heapq.heappop(queue)
print(smallest)  # 출력: 3

 

 

딕셔너리

• items() 딕셔너리의 모든 (키, 값)쌍을 반환하는 함수