📌  相关文章
📜  计算放置在两个给定数组的相同索引处的相同值的元素(1)

📅  最后修改于: 2023-12-03 15:41:39.172000             🧑  作者: Mango

计算两个数组中相同索引处相同值的元素

在编程中,经常需要比较两个数组中相同位置的元素是否相等。本文将介绍计算两个数组中相同索引处相同值的元素的方法。

方法一:使用循环

使用循环遍历两个数组,比较相同索引处的元素是否相等。

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

for i in range(len(arr1)):
    if arr1[i] == arr2[i]:
        result.append(arr1[i])

print(result)

输出:[1, 3]

方法二:使用列表解析

可以使用列表解析快速计算两个数组中相同索引处相同值的元素。

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

result = [arr1[i] for i in range(len(arr1)) if arr1[i] == arr2[i]]

print(result)

输出:[1, 3]

方法三:使用numpy库

numpy库提供了一个where函数,可以返回两个数组中相同索引处相同值的元素下标。

import numpy as np

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

result = np.where(arr1 == arr2)[0]

print(result)

输出:[0 2]

总结

以上就是计算两个数组中相同索引处相同值的元素的方法。可以根据需要选择不同的方法来实现。如果数组较大,建议使用numpy库的方法,因为numpy的运算速度更快。