문제

연오는 개 다섯 마리를 키우고 있다. 어느날 외출하고 돌아오니 누군가가 침대를 물어뜯은 것이 아닌가! 현장 조사 결과, 범인은 주둥이가 작고, 발이 크고, 흰색 털을 가진 개라는 것을 알 수 있었다. 연오는 용의자를 좁히기 위해 개의 정보를 정리했다. 이 데이터를 선별해 용의자로 의심되는 개를 모두 찾아 그 이름을 화면에 출력하라.

용의자 = [
    {'이름': '멍멍', '털': '흰색', '주둥이': '크다', '발': '크다'},
    {'이름': '킁킁', '털': '검은색', '주둥이': '작다', '발': '크다'},
    {'이름': '왈왈', '털': '흰색', '주둥이': '작다', '발': '크다'},
    {'이름': '꿀꿀', '털': '검은색', '주둥이': '작다', '발': '작다'},
    {'이름': '낑낑', '털': '흰색', '주둥이': '작다', '발': '작다'},
]

답

코드

용의자 = [
    {'이름': '멍멍', '털': '흰색', '주둥이': '크다', '발': '크다'},
    {'이름': '킁킁', '털': '검은색', '주둥이': '작다', '발': '크다'},
    {'이름': '왈왈', '털': '흰색', '주둥이': '작다', '발': '크다'},
    {'이름': '꿀꿀', '털': '검은색', '주둥이': '작다', '발': '작다'},
    {'이름': '낑낑', '털': '흰색', '주둥이': '작다', '발': '작다'},
]

# 범인은 주둥이가 작고, 발이 크고, 흰색 털을 가진 개
profiled_dogs_1 = filter(lambda dog: dog['주둥이'] == '작다', 용의자)
profiled_dogs_2 = filter(lambda dog: dog['발'] == '크다', profiled_dogs_1)
profiled_dogs_3 = filter(lambda dog: dog['털'] == '흰색', profiled_dogs_2)

for dog in profiled_dogs_3:
    print(dog['이름'])

실행 결과

왈왈

해설

• 범인은 주둥이가 작고, 발이 크고, 흰색 털을 가진 개다. 따라서, "주둥이가 작다", "발이 크다", "털이 흰색이다"라는 세 조건으로 선별하면 된다.
• 답안에서는 filter() 함수를 세 번 사용하여 세 차례에 걸쳐 선별했다. 물론, filter(lambda dog: dog['주둥이'] == '작다' and dog['발'] == '크다' and dog['털'] == '흰색', 용의자)와 같이 and 연산자로 조건을 합쳐 한번에 선별하는 것도 좋은 방법이다.
• 결과를 출력하면, 선별된 용의자가 한 마리에 불과하다는 걸 알 수 있다. 하지만 그렇다고 하여 선별된 결과 가운데 하나의 요소만 꺼내 출력했다면 오답이다. 문제가 요구하는 바는 선별한 용의자의 이름을 모두 출력하는 것이므로, for 문으로 선별한 용의자의 시퀀스를 순회하여 출력하도록 프로그램을 작성해야 한다. 프로그램을 그와 같이 작성해두어야 나중에 데이터가 변경되더라도 프로그램을 수정 없이 그대로 사용할 수 있다. 즉, 프로그램의 범용성이 커진다.
• filter() 함수 대신 리스트 조건제시법을 사용하는 것도 좋은 방법이다. 다음은 리스트 조건제시법을 이용한 답안의 예다.

print([
    dog['이름']
    for dog
    in 용의자
    if dog['주둥이'] == '작다' and dog['발'] == '크다' and dog['털'] == '흰색'
])

• for 문이나 while 문을 이용해 직접 선별하는 것도 가능하다. for 문이나 while 문은 어떤 경우에도 활용할 수 있는 만능 도구이며, 이것을 자유자재로 활용할 수 있도록 익혀두는 것은 초보 프로그래머가 실력을 향상시키는 좋은 방법이다. 하지만 filter() 함수와 리스트 조건제시법을 사용하는 것도 코드를 더 간결하고 쉽게 작성하는 방법이니 소홀히 하지 말자.