2차원으로 만들기

https://school.programmers.co.kr/learn/courses/30/lessons/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의 길이

 

[나의 풀이]

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

 

[다른 풀이]

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

import numpy as np
def solution(num_list, n):
    li = np.array(num_list).reshape(-1,n)
    return li.tolist()

 

 

예시)

def solution(num_list, n):
    answer = []
    cnt = 0
    temp = []
    for num in num_list:
        temp.append(num)
        cnt += 1
        if cnt == n:
            answer.append(temp)
            temp = []
            cnt = 0

    return answer

 

def solution(num_list, n):
    answer = []
    i = 0
    while i < len(num_list):
        answer.append(num_list[i:i+n])
        i+=n
    return answer