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]
如下与argmax()结合使用可用于投票
y_pred['label'] = y_pred.apply(lambda x:np.argmax(np.bincount(x)), axis=1)
ttt