[Python] python comprehension

Python comprehension ?

다른 Sequence로부터 새로운 Sequence (Iterable Object)를 만들 수 있는 기능
* comprehension :  이해력 

 

 

01. List comprehension

[ 출력 표현식 for 요소 in 입력 Sequence [ if 조건식 ] ]

• 입력 Sequence는 Iteration이 가능한 데이터 Sequence 혹은 컬렉션
• [if 조건식]은 옵션

hash_table = list([0 for i in range(10)])

print(hash_table) #[0, 0, 0, 0, 0, 0, 0, 0, 0, 0]


#1. 종류가 다른 데이터에서 정수 리스트만 가져오기
dataset = [4, True, 'dave', 2.1, 3]
int_data = [num for num num in dataset if type(num) == int]

print(int_data) #[4, 3]

 

---

02. Set comprehension 

{ 출력 표현식 for 요소 in 입력 Sequence [ if 조건식 ] }

• 입력 Sequence로부터 조건에 맞는 새로운 Set 컬렉션 리턴 ( Set : 중복 X )
• [if 조건식]은 옵션
• Python3 부터 추가됨

int_data = [1, 1, 2, 3, 3, 4]

square_data_set = {num * num for num in int_data}
print(square_data_set) #{16, 1, 4, 9}

square_data_set1 = {num * num for num in int_data if num > 3}
print(square_data_set) #{16}

---

03. Dictionary comprehension 

{ Key:Value for 요소 in 입력 Sequence [ if 조건식 ] }

• 입력 Sequence로부터 조건에 맞는 새로운 Dictionary 컬렉션 리턴
• [if 조건식]은 옵션
• Python3 부터 추가됨

id_name = {1:'lee', 2:'kim', 3:'park'}
id_name.items() #dict_items([(1, 'lee'), (2, 'kim'), (3, 'park')])

#아이디가 1보다 큰 데이터, val:key 형식으로
name_id = {val:key for key,val in id_name.items() if key > 1}
print(name_id) #{'kim': 2, 'park': 3}

#아이디 10단위로 바꾸기
name_id = {key*10:val for key,val in id_name.items()}
print(name_id) #{10:'lee', 20:'kim', 30:'park'}