Find the Intersection of Two Arrays in Python

0 min read 154 words

The challenge

Given two arrays, write a function to compute their intersection.

Example 1:

Input: nums1 = [1,2,2,1], nums2 = [2,2]
Output: [2,2]

Example 2:

Input: nums1 = [4,9,5], nums2 = [9,4,9,8,4]
Output: [4,9]

Note:

  • Each element in the result should appear as many times as it shows in both arrays.
  • The result can be in any order.

This is a common array manipulation problem that’s similar to other Python challenges like counting smiley faces or getting the next biggest number with the same digits .

The solution in Python code

def intersect(self, nums1: List[int], nums2: List[int]) -> List[int]:
    # create an `answer` list
    answer = []

    # loop through the first list
    for i in nums1:
        # check if the item is in the next list
        if i in nums2:
            # remove from the nums2 list
            nums2.remove(i)
            # add the answer to the `answer` list
            answer.append(i)

    # return our `answer` list
    return answer
Tags:
Andrew
Andrew

Andrew is a visionary software engineer and DevOps expert with a proven track record of delivering cutting-edge solutions that drive innovation at Ataiva.com. As a leader on numerous high-profile projects, Andrew brings his exceptional technical expertise and collaborative leadership skills to the table, fostering a culture of agility and excellence within the team. With a passion for architecting scalable systems, automating workflows, and empowering teams, Andrew is a sought-after authority in the field of software development and DevOps.

Tags