📅  最后修改于: 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库提供了一个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的运算速度更快。