Python 자료형 공부 2 - array

Python3으로 코딩테스트를 진행하던 중, 잦은 버그가 발생하는 부분과 기억해야할 팁들을 위주로 정리해보고자 한다.

 

1. 2차원 배열 선언

w = 3
h = 5
Matrix = [[0 for x in range(w)] for y in range(h)] 

위와 같이 혹은 다음 아래와 같이 선언이 가능하다. 

h = 5
Matrix = [[0] * w for i in range(h)]

그러나,

Matrix = [[0] * n] * n

위와 같이 선언하게 되면, 각 행은 서로의 행을 복사하였기 때문에 우리가 원하는대로 배열을 사용하기 어렵다. 예를 들면, 1행의 값을 변경시키면 나머지 2 ~ n 행 모두 같은 값으로 변경이 된다.

 

2. 리스트 값 추가

리스트는 append 메소드를 이용하여 값을 추가할 수 있다. 다음과 같이 실행하게 되면, a는 [1,2,3,4] 의 값을 갖게 된다.

a = [1, 2, 3]
a.append(4)

2-1. 2차원 배열 추가

a = []
a.append([])
a[0].append(10)
a[0].append(20)
a.append([])
a[1].append(500)

3. 리스트 위치 반환

index(x) 함수는 리스트에 x 값이 있으면 x 값을 돌려준다. 존재하지 않으면 ValueError가 발생한다.

다음 코드의 결과는 '2'가 나오게된다.

a = [1, 2, 3]
a.index(3)