[프로그래머스 Lv.0] 2차원으로 만들기 _파이썬

문제

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

프로그래머스

 

내 풀이

import numpy as np

def solution(num_list, n):
    return (np.array(num_list).reshape(len(num_list) // n, n)).tolist()

왠지 사용할 수 있는 라이브러리가 있을 것 같아서 찾아보았다.

 

numpy의 reshape함수

np.reshape(변경할 배열, 차원) 또는 배열.reshape(차원) 형식으로 사용하며, 배열의 차원을 변경하여 행렬을 반환하거나 할때 많이 쓰는 함수이다.

 

다른 사람의 풀이

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