Python统计数据集中不同种类的数量

##!/usr/bin/python3
# -*- coding: utf-8 -*-
# @Time    : 2022/03/18 18:10
# @Author  : 任思成
# @File    : count_classes.py

import os
import matplotlib.pyplot as plt

folder_path = "E:/DataSet/XiaoBaoDang_DS/XiaoBaoDang_DS_3_15/XiaoBaoDang_DS_3_15_labels" #存放Labels路径
classes_file = "./classes.txt" #存放类别文件路径

classes_data = []
file_list=os.listdir(folder_path)

for classes in open(classes_file,'r').readlines():
    dr = [classes.replace("\n", ""), 0]
    classes_data.append(dr)

for item in file_list:
    if item.split('.')[1] == "txt" and item != "classes.txt":
        for line in open(folder_path+"/"+item,'r').readlines():
            for i in range(len(classes_data)):
                if line.split(' ')[0] == str(i):
                    classes_data[i][1] += 1
 
fig = plt.figure()
fig.canvas.set_window_title('训练数据类别统计')
plt.rcParams["font.sans-serif"]=['SimHei']
plt.title("训练数据类别统计")
plt.xlabel("类别")
plt.ylabel("数量")
for i in range(len(classes_data)):
    plt.bar(classes_data[i][0], classes_data[i][1])
plt.show()