一、数据类型
每个文件夹下都是这种文件,虽然可以通过手动数出来了解文件数量,但为了更直观地看到每个文件夹的文件数量,可以使用图表来表示,这样会更加清晰。
效果展示:
二、代码实现
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()
标签:count,plt,柱状图,文件夹,file,folder,绘制,counts,255 From: https://blog.csdn.net/weixin_60359250/article/details/139276150