python - 从两个数组创建元素映射

Question

我有两个数组，它们是聚类算法的输出。是否有可能自动找到关联映射。

考虑两个标签数组:

array1 = [0,0,1,2,3]
array2 = [4,4,6,8,7]

视觉上这些看起来是一样的，但对于更大的标签集，我想找到一个像 {0:4,1:6,2:8,3:7} 这样的映射。

Python 有什么方法可以做到这一点吗？

我已经查看了 sklearn metrics 以获取类似的解决方案，但还没有成功。任何帮助将不胜感激。

Answer 1

是的，您可以通过zipping 创建一个字典，其中键作为第一个列表的元素，值作为第二个列表的元素。将两个列表放在一起，并将输出转换为字典

array_1 = [0,0,1,2,3]
array_2 = [4,4,6,8,7]

#Zip the two lists together, and create a dictionary out of the zipped lists
mapping = dict(zip(array_1, array_2))
print(mapping)

输出将是

{0: 4, 1: 6, 2: 8, 3: 7}

请注意，如果您在 array_1 中有重复元素但是array_2中的相应元素不同，array_2的最后一个元素将在映射中为重复元素选择，例如 [0,0,1,1]和 [4,5,6,7] .映射将是 {0: 5, 1: 7} , 自 5为重复元素选择 0和 7为重复元素选择 1