Tuesday_June_30
.
.
.

🗂 List Comprehension

List Comprehension이란, 새로운 리스트를 만들 때 사용할 수 있는 간단한 표현식으로 리스트와 마찬가지로 대괄호를 사용하여 작성한다.

그리고 우리가 만들려고 하는 원소를 표현하는 표현식으로 시작하여 for 루프가 뒤에 따라오는 형식을 가진다. For 문 뒤에 if문을 추가하여 조건문을 포함한 형식도 올 수 있다.

표현 방법

comprehension_list = [x for x in something]

변수 이름 = [변수를 활용한 값 for 사용할 변수 이름 in 반복 가능한 값]

comprehension_list = [x for x in something if ~]

변수 이름 = [변수를 활용한 값 for 사용할 변수 이름 in 반복 가능한 값 if문]

그럼 간단한 예제부터 시작하여 리스트 컴프리헨션의 장점을 살펴 보겠습니다.

위의 리스트 컴프리헨션 형식으로 1부터 10까지를 가지는 리스트를 만들 때 다음과 같이 리스트 컴프리헨션으로 리스트를 생성할 수 있습니다.

new_list = [ x for x in range(1, 11) ]
print(new_list)

1️⃣ 문제

다음과 같은 도시목록의 리스트가 주어졌을때, 도시이름이 S로 시작하지 않는 도시만 리스트로 만들 때 리스트 컴프리헨션을 사용하여 함수를 작성해 보자.

cities = [Tokyo, Shanghai, Jakarta, Seoul, Guangzhou, Beijing, Karachi, Shenzhen, Delhi ]

함수를 정의할 때, parameter로 원래의 cities라는 list를 주고 for문으로 원소들을 받아오는데, 'S'로 시작하는 도시의 이름을 따와야 하니 String에서 첫 글자를 따오려면 슬라이싱을 하면 된다.

cities = ["Tokyo", "Shanhai", "Jakarta", "Seoul",
          "Guangzhou", "Beijing", "Karachi", "Shenzhen", "Delhi"]

print("This is Original List = ")
print(cities)

def pick_s_cities(list):
    list_comprehension = [element for element in list if (element[0]) == "S"]
    return list_comprehension

print("This is Comprehension List = ")
print(pick_s_cities(cities))

출력 결과

This is Original List = 
['Tokyo', 'Shanhai', 'Jakarta', 'Seoul', 'Guangzhou', 'Beijing', 'Karachi', 'Shenzhen', 'Delhi']
This is Comprehension List = 
['Shanhai', 'Seoul', 'Shenzhen']

2️⃣ 문제

다음과 같은 도시, 인구수가 튜플의 리스트로 주어졌을때, 키가 도시 값이 인구수인 딕셔너리를 딕셔너리 컴프리헨션을 사용한 함수를 작성해 보세요.

population_of_city = [("Tokyo", 36923000), ("Shanghai", 34000000), ("Jakarta", 30000000), ("Seoul", 25514000), (
    "Guangzhou", 25000000), ("Beijing", 24900000), ("Karachi", 24300000), ("Shenzhen", 23300000), ("Delhi", 21753486)]

딕셔너리 형태로 만들어주고 반환을 해야하니 일단 list와 다르게 my_dictionary라는 변수를 딕셔너리로 생성하듯 {}로 초기화 해준다. 그리고 위에서 말했듯 List Comprehension의 사용법은 반복문 전에 변수를 활용한 값이 오게된다. 그러므로 우리는 element라는 변수를 활용해 딕셔너리에 값을 주듯 변수를 활용할 것이니 element[0] : element[1] 이렇게 변수를 활용하는 것이다. (참고로 tuple은 수정이 불가능하나 index의 값을 갖고 있는 list와 같아서 앞에 오는 도시의 이름인 0번 째 index를 앞에, 인구수인 1번 째 index를 뒤에 주는 것이다.

population_of_city = [("Tokyo", 36923000), ("Shanghai", 34000000), ("Jakarta", 30000000), ("Seoul", 25514000), (
    "Guangzhou", 25000000), ("Beijing", 24900000), ("Karachi", 24300000), ("Shenzhen", 23300000), ("Delhi", 21753486)]

print("This is Tuples in List")
print(population_of_city)

def make_dictionary(list):
    my_dictionary = {element[0]: element[1] for element in list}
    # my_dictionary = {}
    # for element in list:
    #     my_dictionary[element[0]] = element[1]
    return my_dictionary

print("This is Dictionary")
print(make_dictionary(population_of_city))

출력 결과

This is Tuples in List
[('Tokyo', 36923000), ('Shanghai', 34000000), ('Jakarta', 30000000), ('Seoul', 25514000), ('Guangzhou', 25000000), ('Beijing', 24900000), ('Karachi', 24300000), ('Shenzhen', 23300000), ('Delhi', 21753486)]
This is Dictionary
{'Tokyo': 36923000, 'Shanghai': 34000000, 'Jakarta': 30000000, 'Seoul': 25514000, 'Guangzhou': 25000000, 'Beijing': 24900000, 'Karachi': 24300000, 'Shenzhen': 23300000, 'Delhi': 21753486}