[programmers] 2차원으로 만들기, 120842

문제 설명

정수 배열 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차원 배열로 변경합니다.

 

제한사항

• num_list의 길이는 n의 배 수개입니다.
• 0 ≤ num_list의 길이 ≤ 150
• 2 ≤ n < num_list의 길이

입출력 예

 

입출력 예 설명

입출력 예 #1

• num_list가 [1, 2, 3, 4, 5, 6, 7, 8] 로 길이가 8이고 n이 2이므로 2 * 4 배열로 변경한 [[1, 2], [3, 4], [5, 6], [7, 8]] 을 return합니다.

입출력 예 #2

• num_list가 [100, 95, 2, 4, 5, 6, 18, 33, 948] 로 길이가 9이고 n이 3이므로 3 * 3 배열로 변경한 [[100, 95, 2], [4, 5, 6], [18, 33, 948]] 을 return합니다.

풀이

풀이 #1

#풀이 1
def solution(num_list, n):
    answer = []
    for i in range(len(num_list)//n):
        answer.append(num_list[i*n:(i+1)*n])
    return answer
    
    
#풀이 1-1    
def solution(num_list, n):
    return [num_list[i*n:(i+1)*n] for i in range(len(num_list)//n) ]

풀이 #1 설명

• len(num_list)//n : 만들어지는 2차원 리스트의 개수만큼 반복
• num_list 앞에서 n개씩 2차원 리스트로 저장

풀이 #2

#풀이 2
def solution(num_list, n):
    answer = []
    for i in range(0, len(num_list), n):
        answer.append(num_list[i:i+n])
    return answer
    
#풀이 2-1
def solution(num_list, n):
    return [num_list[i:i+n] for i in range(0,len(num_list),n)]

풀이 #2 설명

• 풀이 1과 동일, range의 범위를 다르게 설정
• range(0, len(num_list), n) : 0부터 num_list길이만큼, n칸씩 건너뛰면서 반복