[Python 문법 정리] 코딩테스트 대비

1. 2차원 배열 선언

array = [[0 for col in range(11)] for row in range(10)]

visited = [[False for _ in range(m)] for _ in range(n)]

array = [[0]*11 for i in range(10)]

array = [[] for _ in range(3)] #인접리스트 구현시 행이 3개인 2차원 벡터로 생각

*주의: 다음 방식으로 선언하면 shallow copy가 되어 매우 위험하므로 안됨!

array = [[0]*11 ]*10

2. 2차원 배열 입력 받기

arr = []
for i in range(m):
  arr.append(list(map(int, input().split())))

3. swap

arr[0], arr[1] = arr[1], arr[0]

4. 정렬

• sorted()

#1
arr = [3,1,4,5]
result = sorted(arr)

#2
arr = [('banana', 2), ('apple', 4), ('kiwi', 9)]

def setting(data):
	return data[1]

result = sorted(arr, key=setting)

• sort()

arr = [3,1,4,5]
arr.sort()
arr.sort(reverse=True)