python/파이썬/지그재그정렬

 

i, j = map(int, input('2차원 리스트를 만들려하는 행과 열의 개수를 입력하세요 : ').split())
data = [[0] * j for i in range(i)]
 
n = 0                    
start = 0               
end = len(data[0]) - 1  
sw = 1                  
 
for i in range(len(data)):
    for j in range(start, end + sw, sw):  
        n += 1
        data[i][j] = n 
    
    start, end = end, start
    sw *= -1
 
for i in range(len(data)):
    for j in range(len(data[0])):
        print('%3d ' % data[i][j], end = '')
 
    print()
 
 
 

결과: