[Programmers] LV 0. 2차원으로 만들기 (Python/파이썬)

▶ 문제 설명

정수 배열 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	result
[1, 2, 3, 4, 5, 6, 7, 8]	2	[[1, 2], [3, 4], [5, 6], [7, 8]]

---

▶ 제한 사항

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

---

▶ 입출력 예

num_list	n	result
[1, 2, 3, 4, 5, 6, 7, 8]	2	[[1, 2], [3, 4], [5, 6], [7, 8]]
[100, 95, 2, 4, 5, 6, 18, 33, 948]	3	[[100, 95, 2], [4, 5, 6], [18, 33, 948]]

---

▶입출력 예 설명

입출력 예 #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합니다.

---

▶ Code

import numpy as np

def solution(num_list, n):
    answer = np.array(num_list).reshape(-1,n)
    return answer.tolist()

▶ Point

numpy array로 변환한 뒤 reshape을 이용했다.

answer의 형이 numpy.ndarray라 tolist()로 list로 형변환해서 return 해준다.

 

reshape() 사용법

numpy의 reshape(x, x)은 reshape(x행, x열)의 2차원 배열로 변경해준다.

l.reshape(2,6)
#[[ 0  1  2  3  4  5]
# [ 6  7  8  9 10 11]]

l.reshape(4,3)
# [[ 0  1  2]
#  [ 3  4  5]
#  [ 6  7  8]
#  [ 9 10 11]]

 

reshape(-1, 정수)는 정수 만큼 열을 생성하고 행을 알아서 지정해준다.

l = np.arange(12)
#l : [0 1 2 3 4 5 6 7 8 9]

l.reshape(-1, 4)
# [[ 0  1  2  3]
#  [ 4  5  6  7]
#  [ 8  9 10 11]]

l.reshape(-1, 6)
# [[ 0  1  2  3  4  5]
#  [ 6  7  8  9 10 11]]

 

reshape(정수, -1)도 마찬가지로 사용가능하다.