统计不同文件夹中的文件数量，并绘制相应的柱状图。

一、数据类型

每个文件夹下都是这种文件，虽然可以通过手动数出来了解文件数量，但为了更直观地看到每个文件夹的文件数量，可以使用图表来表示，这样会更加清晰。

效果展示：

 

 

二、代码实现 

import os
import matplotlib.pyplot as plt

folder_names = ['0', '1', '2', '3']
file_counts = []

# 遍历每个文件夹，获取文件数量
for folder_name in folder_names:
    folder_path = os.path.join('.', folder_name)
    file_list = os.listdir(folder_path)
    file_count = len(file_list)
    file_counts.append(file_count)

# 绘制柱状图
plt.bar(range(len(file_counts)), file_counts, color=(0/255, 127/255, 255/255))
plt.xticks(range(len(file_counts)), folder_names)
plt.xlabel('类别')
plt.ylabel('样本数量')

# 显示每个柱子的数量
for i, count in enumerate(file_counts):
    plt.text(i, count, str(count), ha='center', va='bottom')

plt.title('数量统计')
plt.rcParams['font.sans-serif'] = ['SimHei']  # 指定使用中文字符的字体，如黑体
plt.savefig('bar_chart.png')
plt.show()