반응형
질문
나는 두 개의 목록을 가져와서 두 목록에 모두 나타나는 값을 찾고 싶습니다.
a = [1, 2, 3, 4, 5]
b = [9, 8, 7, 6, 5]
returnMatches(a, b)
예를 들어 [5]
를 반환합니다.
답변
가장 효율적이진 않지만 가장 명백한 방법은 다음과 같습니다:
>>> a = [1, 2, 3, 4, 5]
>>> b = [9, 8, 7, 6, 5]
>>> set(a) & set(b)
{5}
만약 순서가 중요하다면 다음과 같이 리스트 컴프리헨션을 사용할 수 있습니다:
>>> [i for i, j in zip(a, b) if i == j]
[5]
(순서의 중요성을 의미하는 동일한 크기의 리스트에만 작동합니다).
반응형
댓글