Python打印两数组相同元素|极客笔记

Python打印两数组相同元素

在日常编程中，经常会遇到需要比较两个数组中的元素并找出它们的相同元素的情况。Python作为一种简洁而强大的编程语言，提供了多种方法来实现这个需求。本文将详细介绍如何使用Python来打印两个数组中相同的元素。

方法一：使用循环遍历

最直观的方法是利用两层循环来遍历两个数组，然后比较它们的元素。这种方法的时间复杂度为O(n^2)，不适用于大型数组，但对于小型数组却非常直观易懂。

def find_common_elements(arr1, arr2):
    common_elements = []
    for i in arr1:
        for j in arr2:
            if i == j:
                common_elements.append(i)
    return common_elements

arr1 = [1, 2, 3, 4, 5]
arr2 = [3, 4, 5, 6, 7]

result = find_common_elements(arr1, arr2)
print(result)

运行结果：

[3, 4, 5]

方法二：使用集合（Set）

利用Python中的集合（Set）数据结构，可以更高效地找出两个数组中的相同元素。将一个数组转换成集合，然后利用集合的交集操作，即可得到相同的元素。

def find_common_elements(arr1, arr2):
    set1 = set(arr1)
    set2 = set(arr2)
    return list(set1.intersection(set2))

arr1 = [1, 2, 3, 4, 5]
arr2 = [3, 4, 5, 6, 7]

result = find_common_elements(arr1, arr2)
print(result)

运行结果：

[3, 4, 5]

方法三：使用列表推导式

列表推导式是Python中一种简洁而强大的语法，可以在一行代码中完成循环操作。利用列表推导式，我们可以更简洁地找出两个数组中的相同元素。

arr1 = [1, 2, 3, 4, 5]
arr2 = [3, 4, 5, 6, 7]

common_elements = [i for i in arr1 if i in arr2]
print(common_elements)

运行结果：

[3, 4, 5]

方法四：使用集合（Counter）

Python中的collections模块提供了Counter类，可以帮助我们统计每个元素在数组中出现的次数。利用Counter，我们可以更有效地找出两个数组中的相同元素。

from collections import Counter

arr1 = [1, 2, 3, 3, 4]
arr2 = [3, 3, 4, 5, 6]

counter1 = Counter(arr1)
counter2 = Counter(arr2)

common_elements = list((counter1 & counter2).elements())
print(common_elements)

运行结果：

[3, 3, 4]

方法五：使用Numpy

如果你对数组操作较为熟悉，可以使用Python的数值计算库Numpy来完成找出相同元素的操作。Numpy提供了很多强大的数组操作函数，可以帮助我们更高效地处理数组。

import numpy as np

arr1 = np.array([1, 2, 3, 4, 5])
arr2 = np.array([3, 4, 5, 6, 7])

common_elements = np.intersect1d(arr1, arr2)
print(common_elements)

运行结果：