[python] 파이썬 2개의 리스트에서 공통된 값 찾기

Quize
스낵리스트와 초코 리스트가 있을 때 스낵 리스트에 있는 '초코과자'를 찾아라!

---

snack_list = ['꼬북칩', '오사쯔','초코파이','빅파이']
choco_list = ['칙촉','초코파이','다이제','빅파이']

1. 이중 for문 사용

# 1-1 이중 for문
find_choco_snack = []
for snack in snack_list:
    if snack in choco_list:
        find_choco_snack.append(snack)
print(find_choco_snack)

# 1-2 동일내용 comprehension사용하기
find_choco_snack = [snack for snack in snack_list if snack in choco_list]
print(find_choco_snack) 

# ['초코파이', '빅파이'] 동일결과 출력

 

2. set 사용하기

중복과 순서가 없는 데이터구조 set을 사용하며, set & set 으로 교집합을 구함 
set은 '{}'를 사용하여 정의하므로 출력 시 'list'로 변경

find_choco_snack = list(set(snack_list) & set(choco_list))
print(find_choco_snack) # ['초코파이', '빅파이']

3. filter 사용하기

filter()함수로 조건에 맞는 원소를 걸러냄, 해당 문제에서는 2개 리스트에 모두 있는 값을 걸러냄

find_choco_snack = list(filter(lambda snack: snack in choco_list, snack_list))
print(find_choco_snack)