读取CSV文件并用字典统计文件夹下数据数量

import os
import pandas as pd

test_path = 'H:\\50data原始素材\AUTSL\\test_jpgs'
train_path = 'H:\\50data原始素材\AUTSL\\train_jpgs'

# csv_path='H:\\10dataset\\AUTSL\\10AUTSL_test_labels.csv'
csv_path='H:\\10dataset\\AUTSL\\10AUTSL_train_labels.csv'

#读取csv文件
data = pd.read_csv(csv_path)#读取全部数据

file_names=data.iloc[:, 0]#读取第一列
file_num=len(file_names)#统计数量
# print(file_num)

#构建文件夹路径
# i=0
# for file_name in file_names:
#     print(file_name)
#     i+=1
#     if(i==10):
#         break

nums=[]
for file_name in file_names:
    file_name=file_name+'_color'
    file_path = os.path.join(train_path,file_name)
    file_data =os.listdir(file_path)
    image_num=len(file_data)
    nums.append(image_num)
img_num_ = {}
for num in nums:
    if num in img_num_:
        img_num_[num] += 1
    else:
        img_num_[num] = 1

print(img_num_)