python计算众数scipy

计算如下数组中每一行的众数，期望结果，第一行为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]]))
"""