[Python 기초] 파이썬 기본 연습 반복문-2차원 배열 자르기

파이썬 기본 연습  반복문-2차원 배열 자르기

 

파이썬 혼내주기

[문제]

정수를 담은 이차원 리스트, mylist 가 solution 함수의 파라미터로 주어집니다. mylist에 들은 각 원소의 길이를 담은 리스트를 리턴하도록 solution 함수를 작성해주세요.

제한 조건

mylist의 길이는 100 이하인 자연수입니다.
mylist 각 원소의 길이는 100 이하인 자연수입니다.

예시

Input	Output
[[1], [2]]	[1,1]
[[1, 2], [3, 4], [5]]	[2,2,1]

 

내가 풀어본 해답

def solution(mylist):
    answer = []
    for i in mylist:
        answer.append(len(i))
    return answer