[이것이 코딩 테스트다 with Python] 파이썬 문법: 리스트 자료형

리스트 초기화

# 직접 데이터를 넣어 초기화
a = [1, 2, 3, 4, 5, 6, 7, 8, 9]

# 네 번째 원소만 출력
print(a[3])

# 크기가 N이고, 모든 값이 0인 1차원 리스트 초기화
n = 10
a = [0] * 10
print(a)

 

인덱스 값을 입력하여 리스트의 특정한 원소에 접근하는 것을 인덱싱(Indexing)이라고 한다.

• 파이썬의 인덱스 값은 양의 정수와 음의 정수를 모두 사용할 수 있다.
• 음의 정수를 넣으면 원소를 거꾸로 탐색하게 된다.

# 직접 데이터를 넣어 초기화
a = [1, 2, 3, 4, 5, 6, 7, 8, 9]

# 뒤에서 첫 번째 원소 출력
print(a[-1])
# 뒤에서 세 번째 원소 출력
print(a[-3])

# 네 번째 원소 값 변경
a[3] = 7

# 직접 데이터를 넣어 초기화
a = [1, 2, 3, 4, 5, 6, 7, 8, 9]

# 네 번째 원소만 출력
print(a[3]) # 4

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

 

리스트 컴프리헨션

• 리스트를 초기화하는 방법 중 하나
• 대괄호 안에 조건문과 반복문을 적용하여 리스트를 초기화

# 0~9까지의 수를 포함하는 리스트
a = [i for i in range(10)]

# 0~19 까지의 수 중에서 홀수만 포함하는 리스트
array = [i for i in range(20) if i%2 == 1]

# 1부터 9까지의 수들의 제곱 값을 포함하는 리스트
array = [i * i for i in range(1,10)]

• 리스트 컴프리헨션은 2차원 리스트를 초기화할 때 효과적으로 사용될 수 있다.
• 특히 N X M 크기의 2차원 리스트를 한 번에 초기화 해야 할 때 매우 유용하다.

array = [[0]*m for _ in range(n)]

• 파이썬에서는 반복을 수행하되 반복을 위한 변수의 값을 무시하고자 할 때 언더바(_)를 자주 사용한다.

 

리스트 관련 기타 메서드

함수명	사용법	설명	시간 복잡도
append()	변수명.append()	리스트에 원소 삽입	O(1)
sort()	변수명.sort()	기본 정렬 기능 오름차순	O(NlogN)
	변수명.sort(reverse = True)	내림차순 정렬	O(NlogN)
reverse()	변수명.reverse()	리스트 원소 순서 뒤집기	O(N)
insert()	insert(삽입할위치인덱스,삽입할 값)	특정 인덱스 위치에 원소 삽입	O(N)
count()	변수명.count(특정 값)	리스트에서 특정 값을 가지는 데이터의 개수를 셀 때 사용	O(N)
remove()	변수명.remove(특정 값)	특정 값을 갖는 원소를 제거하는데,값을 가진 원소가 여러개면 하나만 제거함	O(N)

 

메서드 실습 코드

a = [4,3,2,1]

# 리스트 원소 뒤집기
a.reverse()
print("원소 뒤집기: ", a) # 원소 뒤집기:  [1, 2, 3, 4]

# 특정 인덱스에 데이터 추가
a.insert(2, 3)
print("인덱스 2에 3 추가: ", a) # 인덱스 2에 3 추가:  [1, 2, 3, 3, 4]

# 특정 값인 데이터 개수 세기
print("값이 3인 데이터 개수: ", a.count(3)) # 값이 3인 데이터 개수:  2

# 특정 값 데이터 삭제
a.remove(3)
print("값이 1인 데이터 삭제: ",a) # 값이 1인 데이터 삭제:  [1, 2, 3, 4]

 

리스트에서 특정 값을 가지는 원소를 모두 제거하기

a = [1,2,3,4,5,5,5]
remove_set = {3, 5} # 집합 자료형

# remove_list에 포함되지 않는 값만을 저장
result = [i for i in a if i not in remove_set]
print(result) # [1, 2, 4]