下面是一个示例代码,用于比较两个3D数组并找到相同的数组:
import numpy as np
# 生成两个随机的3D数组
array1 = np.random.randint(0, 10, size=(3, 3, 3))
array2 = np.random.randint(0, 10, size=(3, 3, 3))
# 找到相同的数组
common_arrays = []
for i in range(array1.shape[0]):
for j in range(array1.shape[1]):
for k in range(array1.shape[2]):
if np.array_equal(array1[i, j, k], array2[i, j, k]):
common_arrays.append(array1[i, j, k])
# 输出结果
if len(common_arrays) > 0:
print("找到了相同的数组:")
for common_array in common_arrays:
print(common_array)
else:
print("没有找到相同的数组")
这个示例代码使用了numpy库来生成随机的3D数组,并使用三个嵌套的for循环遍历数组的每个元素。通过numpy的array_equal
函数来比较两个元素是否相等,如果相等则将其添加到common_arrays
列表中。
最后,如果找到了相同的数组,则将其逐个打印出来;如果没有找到相同的数组,则输出相应的提示信息。