jagged array

길이가 불규칙한 배열을 일정길이로 맞춰주는 코드.

찾아도 안나와서 직접 고안했다. 파이썬을 쓰다보니 for loop를 그냥 쓰는 것은 몬가 불명예스럽다는 느낌이다. 그리고 실제로 더 좋은 방법이 있는 경우가 많은 것 같다.

 

a = [[1],[2,3],[4,5,6]]

# (3, 5) 크기로 변경하고 빈 공간을 0으로 채움

pad = [[0]*i for i in range(1,5)]

b = [ item + pad[4-len(item)]  for item in a]

# 결과 출력

print(b)

 

결과

[[1, 0, 0, 0, 0], [2, 3, 0, 0, 0], [4, 5, 6, 0, 0]]