python - 寻找 numpy 模式向量

Question

我有一个 numpy 数组，其中包含 one-hot 向量。我想找到这些独热向量的众数。请注意，这并不等同于查找值的众数。

例如对于

x = [[0,0,0,1],
     [0,0,0,1],
     [0,0,1,0],
     [0,1,0,0],
     [1,0,0,0]]

assert vector_mode(x) == [0,0,0,1]
assert scipy.stats.mode(x) == [0,0,0,0]

使用 numpy/scipy 执行此操作最有效的方法是什么？

Answer 1

我们将one-hot向量作为2D输入数组的行进行处理。因此，每一行的 argmax 对于每个 one-hot 向量都是唯一的。得到那些。然后，统计他们的数量。具有最大 argmax 计数的任何行都将是所需的模式行输出。让我们选择第一个，再次使用 argmax，最后索引到 2D 输入。

因此，一个实现 -

idx = np.argmax(x,1)
count = np.bincount(idx)
out = x[(idx==count.argmax()).argmax()]