파이썬 기초문법

파이썬에서 이중 리스트로 된 리스트 단일 리스트로 변환

Family in August 2023. 3. 27. 11:55
반응형

파이썬에서 이중 리스트로 된 리스트를 단일 리스트로 변환하는 방법은 flatten() 함수를 사용하는 것입니다.
flatten() 함수는 이중 리스트를 포함한 모든 중첩 리스트를 하나의 리스트로 변환합니다.

아래는 flatten() 함수를 사용한 예시 코드입니다.

def flatten(lst):
    """
    이중 리스트로 된 리스트를 단일 리스트로 변환하는 함수
    """
    result = []
    for item in lst:
        if isinstance(item, list):  # item이 리스트인 경우 재귀적으로 처리
            result.extend(flatten(item))
        else:
            result.append(item)
    return result

# 이중 리스트로 된 리스트
double_list = [[1, 2, 3], [4, 5], [6, 7, 8, 9]]

# 단일 리스트로 변환
single_list = flatten(double_list)

# 출력
print(single_list)  # [1, 2, 3, 4, 5, 6, 7, 8, 9]


위 코드에서 flatten() 함수는 재귀적으로 리스트의 모든 요소를 확인하고, 각 요소가 리스트일 경우에는 다시 flatten() 함수를 호출하여 하나의 리스트로 만듭니다.
리스트가 아닌 경우에는 그대로 결과 리스트에 추가합니다.

이렇게 flatten() 함수를 사용하면 이중 리스트로 된 리스트를 단일 리스트로 변환할 수 있습니다.

반응형