Python에서 2차원 리스트 생성하기

컴프리헨션 내부에 두 개의 반복문을 지정하여 2차원 리스트를 생성할 수 있습니다.

각 내부에 1부터 4까지의 숫자가 들어 있는 세 개의 리스트로 구성된 리스트를 만들어 보겠습니다. 이를 위해서는 컴프리헨션 내부에 또 다른 내부 컴프리헨션이 필요합니다. 그 안에서 변수 j와 반복문을 사용하여 숫자를 생성하겠습니다. 이때 외부 반복문 i는 내부 컴프리헨션 오른쪽에 지정하겠습니다:

pythonlst = [[j for j in range(1, 5)] for i in range(0, 3)]
print(lst)

코드 실행 결과:

python[
  [1, 2, 3, 4],
  [1, 2, 3, 4],
  [1, 2, 3, 4],
]

python[
  [1, 2, 3, 4, 5],
  [1, 2, 3, 4, 5],
  [1, 2, 3, 4, 5],
]

python[
  [
    [1, 2, 3],
    [1, 2, 3],
    [1, 2, 3],
  ],
  [
    [1, 2, 3],
    [1, 2, 3],
    [1, 2, 3],
  ],
  [
    [1, 2, 3],
    [1, 2, 3],
    [1, 2, 3],
  ],
]