python - Numpy 方法返回数组数组中数组出现的索引

Question

我有一个数组数组，表示一组唯一的颜色值:

[[0.         0.         0.        ]
 [0.         0.         1.        ]
 [0.         1.         1.        ]
 [0.5019608  0.5019608  0.5019608 ]
 [0.64705884 0.16470589 0.16470589]
 [0.9607843  0.9607843  0.8627451 ]
 [1.         0.         0.        ]
 [1.         0.84313726 0.        ]
 [1.         1.         0.        ]
 [1.         1.         1.        ]]

另一个 numpy 数组代表其中一种颜色:

[0.9607843  0.9607843  0.8627451 ]

我需要一个函数来查找颜色数组在颜色集中出现的索引，即该函数应该为上面的数组返回 5。

Answer 1

numpy.where() 返回给定条件的值在数组中的确切位置。因此，在这里，它将如下所示(将大数组表示为 arr1，将查找的向量表示为 arr2:

np.where(np.all(arr1 == arr2, axis=1))

然后返回所查找行的行索引数组。