numpy之bincount()

今天在重温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,从而获得统计次数最多的数字



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

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值