以下是Python代码示例,用于比较两个矩阵并找到对应列中匹配的索引:
import numpy as np
matrix1 = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])
matrix2 = np.array([[4, 5, 6], [1, 2, 3], [7, 8, 9]])
matched_index = []
for i in range(matrix1.shape[1]):
for j in range(matrix2.shape[1]):
if np.array_equal(matrix1[:, i], matrix2[:, j]):
matched_index.append((i, j))
print("匹配的索引为:", matched_index)
输出:
匹配的索引为: [(0, 1), (1, 0), (2, 2)]
在这个例子中,我们定义了两个矩阵,并利用嵌套循环遍历它们的列。对每个列,我们利用NumPy库的array_equal
函数检查它们是否匹配。如果找到一对匹配的列,我们将它们的索引存储在元组中。最后,我们输出所有匹配的索引列表。