본문으로 건너뛰기

일차원 배열로 변환하기 해설

일차원 배열로 변환하기 해설

이차원 배열을 일차원 배열로 변환하는 함수를 작성합니다.

리스트 컴프리헨션을 사용하여 이차원 배열의 각 요소를 일차원 배열로 변환합니다.

이는 간결하고 효율적인 한 줄 코드로 구현할 수 있습니다.


모범 답안
def solution(matrix):
return [element for row in matrix for element in row]
  • 리스트 컴프리헨션을 사용하여 각 rowelement를 순서대로 추출하고, 이를 새로운 일차원 배열로 반환합니다.

사용 예시

입출력 예시
print(solution([[1, 2], [3, 4], [5]]))  # 출력: [1, 2, 3, 4, 5]