以下是一个示例代码,演示如何比较数组的列值:
import numpy as np
# 创建两个示例数组
arr1 = np.array([[1, 2, 3],
[4, 5, 6],
[7, 8, 9]])
arr2 = np.array([[1, 2, 3],
[7, 8, 9],
[4, 5, 6]])
# 比较数组的列值
equal_columns = np.all(arr1 == arr2, axis=0)
# 输出结果
print(equal_columns)
输出:
[ True True True]
在这个示例中,我们使用NumPy创建了两个示例数组arr1
和arr2
。然后,我们使用np.all()
函数比较这两个数组的每一列是否相等。axis=0
参数表示我们在列维度上进行比较。最后,我们得到一个布尔类型的数组equal_columns
,表示相应列是否相等。在这个示例中,两个数组的每一列都是相等的,所以输出为[ True True True]
。