일차원 배열로 변환하기 해설
이차원 배열을 일차원 배열로 변환하는 함수를 작성합니다.
리스트 컴프리헨션을 사용하여 이차원 배열의 각 요소를 일차원 배열로 변환합니다.
이는 간결하고 효율적인 한 줄 코드로 구현할 수 있습니다.
모범 답안
def solution(matrix):
return [element for row in matrix for element in row]
- 리스트 컴프리헨션을 사용하여 각
row의element를 순서대로 추출하고, 이를 새로운 일차원 배열로 반환합니다.
사용 예시
입출력 예시
print(solution([[1, 2], [3, 4], [5]])) # 출력: [1, 2, 3, 4, 5]
다음 내용이 궁금하다면?
코드프렌즈 PLUS 멤버십 가입 or 강의를 등록해 주세요!