python (2차원으로 만들기) : 1차원배열 -> 2차원 배열로 쪼개기

도리·2025년 1월 23일

웹서핑하면서 참고한 코드들이다.

  1. answer [ ] 안에 temp [ ] 을 넣기
  • n개수만큼만 temp에 넣어야하기 때문에 n이 될때까지 cnt 하는 변수도 넣어줌.
  • n까지 도달할때 temp와 cnt를 초기화하는 로직
  • temp, cnt 변수를 설정하고 조건에 맞춰 초기화하는 것이 킥인듯 . . .
def solution(num_list, n):
    answer = []
    cnt = 0
    temp = []
    for num in num_list:
        temp.append(num)
        cnt += 1
        if cnt == n:
            answer.append(temp)
            temp = []
            cnt = 0

    return answer
  1. n만큼 자른다. -> n씩 잘라서 n갯수만큼 뒤로자르기.(?)
def solution(num_list, n):
    return [num_list[ix-n:ix] for ix in range(n, len(num_list)+1, n)]
profile
인공지능응용학과 졸업예정..

0개의 댓글