机器学习中分类标签次数统计代码模板
在使用机器学习的算法进行分类时,很多时候我们需要去统计各个分类标签在数据集中出现的次数,它的实现代码模板如下:
def class_count(class_list): # class_list:数据集数据的所有标签列表
# 存放各个分类出现的次数
class_counts = {}
# 统计class_list中每个元素出现的次数
for cla in class_list:
if cla not in class_counts.keys():
class_counts[cla] = 0
class_counts[cla] += 1
# 排序
sorted_class_count = sorted(class_counts.items(), key=operator.itemgetter(1), reverse=True)
return sorted_class_count