파이썬 자료형 - 리스트 자료형 (2)

리스트 컴프리헨션

• 리스트를 초기화하는 방법 중 한 가지!
• 대괄호 [] 안에 조건문과 반복문을 적용하여 리스트를 초기화 할 수 있다.

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

print(array)
# 실행 결과 : [0, 1, 2, 3, 4, 5, 6, 7, 8, 9]

• for i in range(10)
• range(10) : 0부터 9까지
• i의 값이 0부터 9까지 차례대로 순회

• i의 값을 원소로 설정하여 전체를 리스트로 만들겠다.

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

print(array)
# 실행 결과 : [1, 3, 5, 7, 9, 11, 13, 15, 17, 19]


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

print(array)
# 실행 결과 : [1, 4, 9, 16, 25, 36, 49, 64, 81]

• for i in range(1, 10)
• range(1, 10) : 1부터 9까지

• (i * i)의 값을 원소로 설정하여 전체를 리스트로 만들겠다.

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

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

# 일반적인 코드 (리스트 컴프리헨션을 사용하지 않은 코드)
array = []
for i in range(20):
	if i % 2 == 1:
		array.append(i)

• 리스트 컴프리헨션은 2차원 리스트를 초기화할 때 효과적으로 사용 가능하다.
• N x M 크기의 2차원 리스트 한 번에 초기화 하는 경우
• array = [ [0] * m for _ in range(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]]

언더바 _ 언제 사용하나?

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

# "Hello World" 5번 출력하기
for _ in range(5):
	print("Hello World")

리스트 관련 기타 메서드

a = [1, 4, 3]

# 리스트에 원소 삽입
a.append(2)
print(a)       # [1, 4, 3, 2]

# 오름차순 정렬
a.sort()
print(a)       # [1, 2, 3, 4]

# 내림차순 정렬
a.sort(reverse = True)
print(a)       # [4, 3, 2, 1]

a = [4, 3, 2, 1]

# 원소의 순서 뒤집기
a.reverse()
print(a)             # [1, 2, 3, 4]

# 특정한 인덱스 위치에 원소 삽입
a.insert(2, 3)
print(a)             # [1, 2, 3, 3, 4]

# 특정 값 가지는 데이터 개수
print(a.count(3))    # 2

# 특정 값 가지는 원소 제거
a.remove(1)
print(a)             # [2, 3, 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]

이 글은 유튜브 “동빈나” 채널의 “(이코테 2021 강의 몰아보기) 1. 코딩 테스트 출제 경향 분석 및 파이썬 문법 부수기” 영상을 보고 작성하였습니다.