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