下面是一个使用array_positions()函数比较数组而不是等式的代码示例:
import numpy as np
def array_positions(arr1, arr2):
positions = np.where(arr1 == arr2)
return positions
# 例子
arr1 = np.array([1, 2, 3, 4, 5])
arr2 = np.array([2, 4, 6, 8, 10])
positions = array_positions(arr1, arr2)
print("Array positions:", positions)
这个例子中,我们定义了一个array_positions()函数,它接受两个数组作为参数。函数内部使用np.where()函数来找到arr1和arr2中相等元素的位置,并将这些位置存储在positions变量中。最后,我们打印出positions的值。
输出:
Array positions: (array([1, 3]),)
这意味着在arr1和arr2中,索引1和索引3处的元素是相等的。