在numpy数组中查找模式的最有效方法

在numpy数组中查找模式的最有效方法

检查

scipy.stats.mode()

（受@
tom10的评论启发）：

import numpy as npfrom scipy import statsa = np.array([[1, 3, 4, 2, 2, 7],   [5, 2, 2, 1, 4, 1],   [3, 3, 2, 2, 1, 1]])m = stats.mode(a)print(m)

输出：

ModeResult(mode=array([[1, 3, 2, 2, 1, 1]]), count=array([[1, 2, 2, 2, 1, 2]]))

如您所见，它既返回模式又返回计数。您可以直接通过

m[0]

以下方式选择模式：

print(m[0])

输出：

[[1 3 2 2 1 1]]