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

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

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

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

복사
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]