[프로그래머스/파이썬] 2차원으로 만들기

[문제]

정수 배열 num_list와 정수 n이 매개변수로 주어집니다. 
num_list를 다음 설명과 같이 2차원 배열로 바꿔 return하도록 solution 함수를 완성해주세요.

num_list가 [1, 2, 3, 4, 5, 6, 7, 8] 로 길이가 8이고 n이 2이므로 
num_list를 2 * 4 배열로 다음과 같이 변경합니다. 
2차원으로 바꿀 때에는 num_list의 원소들을 앞에서부터 n개씩 나눠 2차원 배열로 변경합니다.

[답안]

def solution(num_list, n):
    return [num_list[i:i + n] for i in range(0, len(num_list), n)]

[풀이]

시작값 0, 마지막 값 len(num_list): num_list의 길이, 스텝 n 으로 for 문을 돌렸다.

num_list의 i 부터 i+n 값 까지 값을 리턴하였다.

예를들어 i = 0, n = 2 일 경우에 index 0부터 2까지 세개의 값을 가지는 1차원 배열이 리턴된다.

이를 for문이 끝날 때 까지 리턴하면 2차원 배열 값을 리턴할 수 있다.

 

[문제 풀어보기]

https://school.programmers.co.kr/learn/courses/30/lessons/120842