Python - 두 리스트에서 공통 요소 값 찾기

pythonlist

두개의 리스트가 갖고 있는 요소 값들 중에서, 서로 공통적으로 갖고 있는 요소를 찾는 방법에 대해서 알아보겠습니다.

1. Set의 intersection()을 이용한 방법

set에는 intersection()이라는 함수가 있는데, 두개의 Set에서 공통 요소를 찾아서 set로 리턴하는 함수입니다.

아래와 같이 리스트를 Set로 변환하고 intersection()으로 공통 요소를 찾고, 다시 리스트로 변환하는 방법이 있습니다.

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

common_elements_set = set(list1).intersection(list2)
common_elements_list = list(common_elements_set)
print(common_elements_list)

Output:

[4, 5]

& 연산자를 이용한 방법

Set는 &를 이용하여 아래와 같이 intersection() 함수와 동일한 동작을 수행하도록 만들 수 있습니다. 오직 Set 자료형에서만 &로 이러한 동작을 하기 때문에, 리스트를 먼저 Set로 변환하고 사용해야 합니다.

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

common_elements_set = set(list1) & set(list2)
common_elements_list = list(common_elements_set)
print(common_elements_list)

Output:

[4, 5]

2. List comprehension을 이용한 방법

리스트 컴프리헨션으로 리스트를 순회하면서 list1에 있는 요소가 list2에 있는지 확인해보면서 공통 요소를 찾을 수 있습니다.

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

common_elements = [element for element in list1 if element in list2]
print(common_elements)

Output:

[4, 5]
Loading script...

Related Posts

codechachaCopyright ©2019 codechacha