List Comprehension
List를 간단하게 생성해주는 함수
표현식 + for문
result = [ 표현식
for변수in리스트 ]
n = 5
# n개의 0으로 초기화된 리스트
result = [ 0 for i in range(n) ] # [0, 0, 0, 0, 0]
# 0 ~ n으로 초기화된 리스트
result = [ i for i in range(n) ] # [0, 1, 2, 3, 4]
# array 리스트의 값을 그대로 복사
array = [ 0, 1, 2, 3, 4, 5, 6, 7, 8, 9 ]
result = [ n for n in array ] # [0, 1, 2, 3, 4, 5, 6, 7, 8, 9]
# array의 제곱값을 구하는 리스트
result = [ n*n for n in array ] # [0, 1, 4, 9, 16, 25, 36, 49, 64, 81]
# 문자열 리스트의 각 문자열의 길이를 저장하는 리스트
str_array = [ "List", "Comprehension", "python"]
result = [ len(string) for string in str_array] # [4, 13, 6]
# 5로 나눈 나머지를 저장하는 리스트 - 함수 사용
def mod_5 (number) :
return number % 5
result = [mod_5(n) for n in array] # [0, 1, 2, 3, 4, 0, 1, 2, 3, 4]표현식 + for문 + 조건문
result = [ 표현식
for변수in리스트if조건문 ]
- 단일 조건문만 사용 가능 ( else, elif 불가 )
- if를 여러번 사용하는 것은 가능 ( and 와 같음 )
# 범위 내 짝수만 저장하는 리스트
result = [ n for n in range(10) if n%2 == 0 ] # [0, 2, 4, 6, 8]
# 범위 내 홀수만 저장하는 리스트
result = [ n for n in range(10) if n%2 != 0 ] # [1, 3, 5, 7, 9]
# 배열에서 양수만 저장하는 리스트
array = [ -1, 0, -4, 24, 5, -10, 2 ]
result = [ n for n in array if n > 0 ] # [24, 5, 2]
##### 다중 if #####
# 3의 배수 중 홀수만 저장하는 리스트
result = [ n for n in range(50) if n%3 == 0 if n%2 != 0 ]
# [3, 9, 15, 21, 27, 33, 39, 45]
result = [ n for n in range(50) if n%3 == 0 and n%2 != 0 ]
# [3, 9, 15, 21, 27, 33, 39, 45]표현식 + 조건문 + for문
result = [ 표현식
if조건문else표현식for변수in리스트 ]
# 양수는 그대로, 음수는 0으로 저장하는 리스트
array = [ -1, 0, -4, 24, 5, -10, 2 ]
result = [ n if n>0 else 0 for n in array ] # [0, 0, 0, 24, 5, 0, 2]
# 짝수라면 'even', 홀수라면 'odd'를 저장하는 리스트
array = [0, 1, 2, 3]
result = [ 'even' if n%2== 0 else 'odd' for n in array ]
# ['even', 'odd', 'even', 'odd']중첩 for문
result = [ 표현식
for변수in리스트for변수 in 리스트 ]
# 중첩 for문
pos = []
for i in range(1, 4) :
for j in range(1, 3) :
pos.append(i*j)
pos = [ i*i for i in range(1, 4) for j in range(1, 3) ] # [1, 2, 2, 4, 3, 6]중첩 List Comprehension
result = [ [ 표현식
for변수in리스트 ]for변수 in 리스트 ]
# 중첩 list comprehension으로 2차원 배열 만들기
result = [ [ 0 for i in range(2) ] for j in range(3) ] # [ [0, 0], [0, 0], [0, 0] ]정렬 (Sort)
- Sort
내부적으로 정렬되어 원본에 저장
- Sorted
원본은 건드리지 않고 정렬된 원소들을 새로운 객체에 담아서 반환
numbers = [-1. 3. -4, 5, 6, 100]
lists = sorted(numbers, key=abs) # 절대값(abs) 순서대로 정렬
lists_2nd = sorted(numbers) # 기본값(작은 순서) 대로 정렬
# key 값에는 함수(function)가 들어감
