numpy之bincount()

来源:互联网 发布:工地记工软件 编辑:程序博客网 时间:2024/05/17 05:05

今天在重温cs231n的assignment1,KNN需要统计K个label中出现频率最高的,label保存在数组中,因此简化为统计数组中出现次数最多的元素,并且返回。

此处使用了numpy中的bincount():https://docs.scipy.org/doc/numpy/reference/generated/numpy.bincount.html

bincount()统计array中非负整数的出现次数,其中value作为返回array的索引,索引对应的值表示value的统计次数:out[value]=occurrences

>>> np.bincount(np.arange(5))array([1, 1, 1, 1, 1])>>> np.bincount(np.array([0, 1, 1, 3, 2, 1, 7]))array([1, 3, 1, 1, 0, 0, 0, 1])

之后再使用argmax()可以返回最大值的索引,此索引刚好对应的是value,从而获得统计次数最多的数字



恩,一直没有写东西的习惯,很多函数都是在要用时才会去查,感觉浪费了不少时间,其中有些是之前使用过的方法,但是由于间隔较远也忘记了必须再上网查询,因此决定把一些觉得有用的东西记下来,方便日后查询。