파이썬 - 리스트 컴프리헨션, insert() , append() , remove() ..

리스트 컴프리헨션

리스트를 초기화하는 방법 중 하나이며, 대괄호([]) 안에 조건문과 반복문을 적용하여 리스트를 초기화할 수 있습니다.

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

출력

[0, 1, 2, 3, 4, 5, 6, 7, 8, 9]

 

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

 , 특히 N X M 크기의 2차원 리스트를 한 번에 초기화해야 할 때 매우 유용하다.

 

- 좋은 예시: array = [[0] * m for _ in range(n)]

- 만약 2차원 리스트를 초기화할 때 다음과 같이 작성하면 예기치 않은 결과가 나올 수 있다.

   > 잘못된 예시: array = [[0] * m] * n

      위 코드는 전체 리스트 안에 포함된 각 리스트가 모두 같은 객체로 인식된다.

 

리스트 컴프리헨션 - 좋은 예시

# N X M 크기의 2차원 리스트 초기화
n = 4
m = 3
array = [[0] * m for _ in range(n)]
print(array)

출력

[[0, 0, 0], [0, 0, 0], [0, 0, 0], [0, 0, 0]]

 

리스트 컴프리헨션 - 잘못된 예시

# N X M 크기의 2차원 리스트 초기화 (잘못된 방법)
n = 4
m = 3
array = [[0] * m] * n
print(array)

array[1][1] = 5
print(array)

출력

[[0, 0, 0], [0, 0, 0], [0, 0, 0], [0, 0, 0]]
[[0, 5, 0], [0, 5, 0], [0, 5, 0], [0, 5, 0]]

 

리스트 컴프리헨션 과 일반적인 코드 비교

리스트 컴프리헨션

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

print(array)

 

일반적인 코드

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

print(array)

출력 결과 동일하다.

[1, 3, 5, 7, 9, 11, 13, 15, 17, 19]

 

2차원 리스트를 초기화할 때 매우 효과적으로 사용할 수 있다.

# N X M 크기의 2차원 리스트 초기화
n = 3
m = 4
array = [[0] * m for _ in range(n)]
print(array)

출력

[[0, 0, 0, 0], [0, 0, 0, 0], [0, 0, 0, 0]]

 

언더바(_)는 어떤 역할일까요?

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

 

# 1부터 9까지의 자연수를 더하기

summary = 0
for i in range(1, 10):
    summary += i
print(summary)

 

# "Hello World"를 5번 출력하기

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

보시면 "Hello World"는 반복을 위한 변수의 값이 필요가 없습니다. 그렇기 때문에 언더바(_)를 사용한다.

 

리스트 관련 기타 메서드

함수 명사 용법 설명 시간 복잡도

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

 

# 리스트 관련 기타 메서드
a = [1, 4, 3]
print("기본 리스트: ", a)

# 리스트에 원소 산입
a.append(2)
print("삽입: ", a)

# 오름차순 정령
a.sort()
print("오름차순 정령: ", a)

# 내림차순 정령
a.sort(reverse=True)
print("내림차순 정렬: ", a)

# 리스트 원소 뒤집기
a.reverse()
print("원소 뒤집기: ", a)

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

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

# 특정 값 데이터 삭제
a.remove(1)
print("갑이 1인 데이터 삭제 :", a)

출력

기본 리스트:  [1, 4, 3]
삽입:  [1, 4, 3, 2]
오름차순 정령:  [1, 2, 3, 4]
내림차순 정렬:  [4, 3, 2, 1]
원소 뒤집기:  [1, 2, 3, 4]
인덱스에 2에 3 추가: [1, 2, 3, 3, 4]
값이 3인 데이터 개수:  2
갑이 1인 데이터 삭제 : [2, 3, 3, 4]

이 중에서 insert() 함수와 append(), remove()를 특이 눈여겨 둬야 한다.

 

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

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]

result = [i for i in a if i not in remove_set]의 의미는 a에 포함된 원소를 하나씩 확인하여 그 원소가 remove_set에 포함되어 있지 않을 때만 리스트 변수인 result에 넣겠다는 의미이다.

---

오늘은 리스트 컴프리헨션과 리스트 관련 기타 메소드들에 대해 공부해 보았습니다.

많은 도움이 되길 바라면 저는이만..