计算如下数组中每一行的众数,期望结果,第一行为1,第二行为3
[[0, 0, 1, 1, 1],
[2, 3, 3, 2, 3]]
使用scipy
的统计包stats
中的mode
函数,指定第二个维度,返回元组,第一个是众数,第二个是出现次数统计。
from scipy import stats
import numpy as np
arr = np.array([[0, 0, 1, 1, 1], [2, 3, 3, 2, 3]])
res = stats.mode(arr, axis=1)
print(res)
"""
ModeResult(mode=array([[1],
[3]]), count=array([[3],
[3]]))
"""