리스트 컴프리헨션, 집합 자료형

리스트 컴프리헨션

- 리스트를 초기화하는 방법 중 하나

- 대괄호 안에 조건문과 반복문을 넣는 방식으로 리스트를 초기화할 수 있음

 

[ 0부터 19까지의 수 중 홀수만 포함하는 리스트 만들기 ]

arr = [i for i in range(20) if i % 2==1]
print(arr) #[1, 3, 5, 7, 9, 11, 13, 15, 17, 19]

 

 

리스트 컴프리헨션은 2차원 리스트를 초기화할 때 효과적으로 사용

n=3
m=4
arr = [[0] * m for _ in range(n)]
print(arr) #[[0, 0, 0, 0], [0, 0, 0, 0], [0, 0, 0, 0]]

 

특정 크기의 2차원 리스트를 초기화할 때에는 반드시 리스트 컴프리헨션을 이용해야 함

 

다음과 같이 2차원 리스트를 초기화한다면, 의도하지 않은 결과가 나올 수 있음

n = 3
m = 4
a = [[0] * m] * n
print(a) # [[0, 0, 0, 0], [0, 0, 0, 0], [0, 0, 0, 0]]

a[1][1] = 5
print(a) # [[0, 5, 0, 0], [0, 5, 0, 0], [0, 5, 0, 0]]

a[1][1]의 값을 5로 바꾸었는데 3개의 리스트에서 인덱스 1에 해당하는 원소의 값이 모두 5로 바뀐 것을 확인할 수 있음

이는 내부적으로 포함된 3개의 리스트가 모두 동일한 객체에 대한 3개의 레퍼런스로 인식되기 때문

 

+) 언더바 (_)의 역할

파이썬에서는 반복을 수행하되 반복을 위한 변수의 값을 무시하고자 언더바를 사용함

단순히 'Hello World'를 5번 출력할 때는 언더바를 사용해 무시할 수 있음

for _ in range(5):
	print('Hello World')

 

 

리스트 관련 기타 메서드

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

 

- insert() 함수를 사용할 때 원소의 개수가 N개면 시간복잡도는 O(N)임

- remove()의 시간복잡도도 O(N)임을 유의. insert()와 마찬가지로 리스트 중간에 있는 우너소를 삭제한 뒤에, 리스트의 원소 위치를 조정해주어야 하기 때문에 O(N)이 소요됨

- 특정한 값의 원소를 모두 제거하려면 remove_all()과 같은 함수로 특정 값의 모든 원소를 제거할 수 있음

- 하지만 파이썬은 이러한 함수를 기본적으로 제공해주지 않으므로 다음과 같은 방법을 이용

a = [1, 2, 3, 4, 5, 5, 5]
remove_set = {3, 5} #제거할 값

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

---

집합 자료형 (set)

- 하나의 집합 데이터에 값을 추가할 때는 add() 함수를 이용할 수 있음

- update() 함수는 여러 개의 값을 한꺼번에 추가하고자할 때 사용

- remove() 함수는 특정 값을 제거할 때 사용

- add(), remove() 함수의 시간 복잡도 O(1)

 

 

 

참고 서적

이것이 취업을 위한 코딩 테스트다 with 파이썬