2차원으로 만들기 - Python, 리스트 컴프리헨션, 2차원

 

 

https://school.programmers.co.kr/tryouts/85906/challenges?language=python3

 

 

 

풀이 - 나의 생각

이차원 리스트 쓰는게 생각보다 불편했다.

 

for문 1개 쓰는 방법, 2개 쓰는 방법이 있다(그 외에도 있겠지만)

 

1개 쓸때는 슬라이드 방식으로 하고,

 

2개 쓸때는 그냥 num_list의 index를 계산해서 넣어주면 된다.

 

코드

 

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

 

 

 

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

 

 

 

다음에 또 봐요