[프로그래머스] 배열 원소의 길이

문자열 배열 strlist가 매개변수로 주어집니다. strlist 각 원소의 길이를 담은 배열을 retrun하도록 solution 함수를 완성해주세요.

def solution(strlist):
answer = []

for i in strlist:
    answer.append(len(i))
return answer

def solution(strlist):
return [len(str) for str in strlist]