Computer >> 컴퓨터 >  >> 프로그래밍 >> Python

파이썬에서 중첩 리스트(리스트의 리스트) 길이 구하는 방법

파이썬에서 중첩 리스트의 전체 요소 개수 구하기

리스트 안에 리스트가 포함된 중첩 리스트(nested list) 구조에서는 단순히 len() 함수만으로는 전체 요소 개수를 알 수 없습니다. 이때 중첩 반복문을 사용하면 각 하위 리스트의 요소를 하나씩 순회하면서 총 개수를 셀 수 있습니다.

1. 중첩 반복문을 사용하는 방법

바깥쪽 반복문으로 각 하위 리스트를 꺼내고, 안쪽 반복문으로 그 안의 요소를 하나씩 확인하며 카운트를 증가시킵니다.

>>> a = [[1, 2, 3], [4, 5, 6]]
>>> c = 0
>>> for x in a:
      for y in x:
          c = c + 1

>>> c
6

위 예제에서는 하위 리스트 두 개에 각각 3개씩 요소가 있으므로, 최종 결과는 6이 됩니다.

2. len() 함수를 조합한 간결한 방법

반복문 없이 더 파이썬다운 방식으로 작성하려면 sum()과 제너레이터 표현식을 함께 사용할 수 있습니다.

>>> a = [[1, 2, 3], [4, 5, 6]]
>>> total = sum(len(x) for x in a)
>>> total
6

이 방법은 각 하위 리스트의 길이를 len()으로 구한 뒤 모두 더하는 방식으로, 코드가 짧고 가독성이 좋아 실무에서 널리 사용됩니다.

정리

  • 중첩 반복문: 요소를 직접 순회하며 개수를 세는 기본적인 방법
  • sum(len(x) for x in a): 간결하고 효율적인 한 줄 코드

상황에 따라 적절한 방법을 선택하면 중첩 리스트의 전체 길이를 손쉽게 구할 수 있습니다.