1. quiz

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

2. answer

import numpy as np

def solution(num_list, n):
    return (np.array(num_list).reshape((len(num_list) // n ,n))).tolist()
  • list를 np.array를 사용하여 배열 형태를 바꿔주었고, 그걸 다시 tolist()를 사용하여 list 형태로 바꿔주었다.

3. 다른 사람의 풀이

def solution(num_list, n):
    return [num_list[ix-n:ix] for ix in range(n, len(num_list)+1, n)]
profile
To be a changer who can overturn world

0개의 댓글