我想比较两个numpy数组并创建第三个数组

import numpy as np

# 创建两个示例数组
arr1 = np.array([1, 2, 3, 4, 5])
arr2 = np.array([5, 4, 3, 2, 1])

# 比较两个数组并创建第三个数组
result = arr1 == arr2

print("arr1:", arr1)
print("arr2:", arr2)
print("result:", result)

arr1: [1 2 3 4 5]
arr2: [5 4 3 2 1]
result: [False False  True False False]

# 确保两个数组的形状相同
arr1 = np.array([1, 2, 3])
arr2 = np.array([1, 2, 3, 4])  # 形状不匹配

# 这将引发 ValueError
result = arr1 == arr2

# 调整数组形状使其匹配
arr2 = arr2[:3]  # 截取前三个元素

result = arr1 == arr2
print("result:", result)

# 确保两个数组的数据类型相同
arr1 = np.array([1, 2, 3], dtype=np.int32)
arr2 = np.array([1, 2, 3], dtype=np.float64)  # 数据类型不匹配

# 这将引发 TypeError
result = arr1 == arr2

# 转换数据类型使其匹配
arr2 = arr2.astype(np.int32)

result = arr1 == arr2
print("result:", result)