2차원으로 만들기

송용진·2023년 8월 29일
0

알고리즘

목록 보기
91/173

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

내 코드

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

예시 코드1

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

예시 코드2

def solution(num_list, n):
    return [num_list[ix-n:ix] for ix in range(n, len(num_list)+1, n)]

예시 코드3

def solution(num_list, n):
    return [num_list[ix-n:ix] for ix in range(n, len(num_list)+1, n)]
profile
백엔드 개발자

0개의 댓글