np.bincount()函数

1 该函数用于统计一个非负的list或array中元素的出现次数

import numpy as np
x = np.array([0, 1, 1, 10])
# bincount()内可以是numpy,也可以是list,注意里面的数必须是非负数,否则报错
# 如果里面有n个数,则输出一个长度为n+1的numpy,第i个索引对应的数即i出现的次数
print(np.bincount(x))
print(np.bincount([0, 1, 1, 3, 2, 1, 5]))
# [1 2 0 0 0 0 0 0 0 0 1]
# [1 3 1 1 0 1]

View Code

如下与argmax()结合使用可用于投票

y_pred['label'] = y_pred.apply(lambda x:np.argmax(np.bincount(x)), axis=1)

ttt