Lv.0_2차원으로 만들기

dowon 2023. 6. 16. 21:43

프로그래머스

코드 중심의 개발자 채용. 스택 기반의 포지션 매칭. 프로그래머스의 개발자 맞춤형 프로필을 등록하고, 나와 기술 궁합이 잘 맞는 기업들을 매칭 받으세요.

programmers.co.kr

처음에 푼 코드

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

단순히 슬라이스를 활용하면 되는 문제였다

하지만, 뭔가 numpy를 쓰고 싶어서 찾아본 결과

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

이런식으로도 풀 수 있었다