Python比较两个等长数组并计算相同元素比例

使用Python NumPy高效比较两个等长数组并计算相同元素比例

如果您需要比较两个等长数组并计算相同元素所占的比例，可以使用NumPy的布尔索引来高效实现。

以下是用Python代码实现的示例：

def count_ratio(array1, array2):    """计算两个等长数组中相同元素所占的比例。

    Args:        array1: 第一个数组。        array2: 第二个数组。

    Returns:        相同元素所占的比例。    """    count = np.sum(array1 == array2)    ratio = count / len(array1)    return ratio

# 示例数据array1 = np.array([1, 2, 3, 4, 5])array2 = np.array([1, 4, 2, 5, 3])

# 统计相同元素所占比例ratio = count_ratio(array1, array2)print("相同元素比例: ", ratio)

在这个代码中，我们定义了一个名为 `count_ratio` 的函数，它接受两个等长的数组作为参数。函数内部使用 `np.sum` 函数对两个数组进行条件判断，统计两个数组中相同元素的个数。然后，我们计算相同元素的个数占总元素数量的比例，并返回结果。

在示例中，我们将两个示例数组 `array1` 和 `array2` 作为输入，并调用 `count_ratio` 函数来统计相同元素所占的比例，最后打印输出结果。

这个方法利用了NumPy高效的布尔索引和数组运算，可以快速地完成比较和计算任务，适用于处理大规模数据。