들어가기 전 문제

문제

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

풀이

이번 문제는 파이썬을 효율적으로 이용해 문제를 푸는것이 목표이기에 파이썬의 함수를 최대한 이용하는 것이 중요하다.

def solution(mylist):
	return list(map(len,mylist))

map 함수를 이용해 모든 인덱스의 길이를 리턴해주게 된다.