[프로그래머스] 서울에서 김서방 찾기Lv.1 - Python

[프로그래머스] 서울에서 김서방 찾기Lv.1

나의 풀이

def solution(seoul):
    return "".join([f"김서방은 {i}에 있다" for i in range(len(seoul)) if seoul[i] == "Kim"])

• 입력된 배열을 돌면서 만약 이름이 Kim인 서방이 있다면, "김서방은 x에 있다" 리스트가 만들어진다. 이 리스트를 join 을 사용해서 문자열로 바꿔서 리턴