실습 : 두 개의 리스트에서 공통 요소 찾기

두 개의 리스트에서 공통 요소 찾기

두 개의 리스트에서 같은 값을 가지는 요소만 추출해서 리스트로 반환하는 common_elements 함수를 작성합니다. 

 

list1 = [1, 2, 3, 4, 5, 6]
list2 = [4, 5, 6, 7, 8, 9]

list1과 list2라는 두 개의 리스트를 정의합니다. 각 리스트에는 여러 개의 정수가 포함되어 있습니다. 

 

common = common_elements(list1, list2)
print(common)       # [4, 5, 6]

common_elements 함수를 호출하면 list1과 list2의 공통 요소를 찾아내어 그 결과를 common 변수에 저장합니다. print(common) 을 통해 common 변수에 저장된 공통 요소들([4, 5, 6])을 출력합니다. 

 

def common_elements(lst1, lst2):
    return list(set(lst1) & set(lst2))

common_elements 함수는 두 개의 리스트 lst1과 lst2를 입력으로 받습니다.
set(lst1) & set(lst2) 부분은 두 리스트를 집합으로 변환한 후, 두 집합의 교집합을 구합니다. 집합(set)은 중복된 요소를 허용하지 않으며, 교집합(&) 연산자는 두 집합 간의 공통 요소만을 반환합니다.
교집합 연산 결과인 집합(set)을 list() 함수를 이용해 다시 리스트로 변환해 반환합니다.

 

전체 코드는 다음과 같습니다. 

def common_elements(lst1, lst2):
    return list(set(lst1) & set(lst2))

list1 = [1, 2, 3, 4, 5, 6]
list2 = [4, 5, 6, 7, 8, 9]

common = common_elements(list1, list2)
print(common)       # [4, 5, 6]