在numpy数组中查找最常见的子数组

Question

示例数据:

array(
  [[ 1.,  1.],
   [ 2.,  1.],
   [ 0.,  1.],
   [ 0.,  0.],
   [ 0.,  0.]])

期望的结果

>>> [0.,0.]

ie)最常见的一对.

看似不起作用的方法:

使用statisticsnumpy数组是不可用的.

使用scipy.stats.modeas作为返回每个轴上的模式,例如,它给出了我们的示例

mode=array([[ 0.,  1.]])

Answer 1

您可以numpy使用该unique功能有效地执行此操作:

pairs, counts = np.unique(a, axis=0, return_counts=True)
print(pairs[counts.argmax()])

返回: [ 0. 0.]