python如何读取文件名

原创

mob64ca12dfd1d5 2023-09-19 10:50:50 ©著作权

文章标签 文件名 Python 文件操作函数 文章分类 Python 后端开发

©著作权归作者所有：来自51CTO博客作者mob64ca12dfd1d5的原创作品，请联系作者获取转载授权，否则将追究法律责任

项目方案：Python文件名读取工具

1. 项目背景

在日常开发中，我们经常需要读取文件夹下的所有文件名，并对其进行相应的处理。比如，我们需要统计某个文件夹中不同类型文件的数量，或者根据文件名的特征来进行筛选等操作。Python作为一种强大的编程语言，提供了丰富的文件操作函数，可以轻松实现文件名的读取。

本项目旨在开发一款简单易用的Python文件名读取工具，方便用户快速获取文件夹下的所有文件名，并进行进一步的处理。

2. 方案概述

本项目将使用Python编程语言，利用Python的文件操作函数来实现文件名的读取。具体方案如下：

用户输入一个文件夹的路径作为输入参数。
程序读取该文件夹下的所有文件名，并保存到一个列表中。
用户可以选择使用正则表达式对文件名进行筛选，并将符合条件的文件名保存到另一个列表中。
用户还可以选择统计不同类型文件的数量，并以饼状图的形式展示。
最后，用户可以选择将最终结果保存到一个文本文件中。

3. 代码实现

下面是该项目的核心代码实现：

import os
import re
import matplotlib.pyplot as plt

def get_all_filenames(folder_path):
    filenames = []
    for root, dirs, files in os.walk(folder_path):
        for file in files:
            filenames.append(file)
    return filenames

def filter_filenames(filenames, pattern):
    filtered_filenames = []
    for filename in filenames:
        if re.search(pattern, filename):
            filtered_filenames.append(filename)
    return filtered_filenames

def count_file_types(filenames):
    file_types = {}
    for filename in filenames:
        file_type = filename.split('.')[-1]
        if file_type in file_types:
            file_types[file_type] += 1
        else:
            file_types[file_type] = 1
    return file_types

def plot_pie_chart(file_types):
    labels = file_types.keys()
    sizes = file_types.values()
    plt.pie(sizes, labels=labels, autopct='%1.1f%%')
    plt.axis('equal')
    plt.show()

def save_results(filenames, filtered_filenames, file_types):
    with open('results.txt', 'w') as f:
        f.write('All Filenames:\n')
        for filename in filenames:
            f.write(filename + '\n')
        f.write('\nFiltered Filenames:\n')
        for filename in filtered_filenames:
            f.write(filename + '\n')
        f.write('\nFile Type Counts:\n')
        for file_type, count in file_types.items():
            f.write(file_type + ': ' + str(count) + '\n')

4. 使用示例

下面是一个简单的使用示例：

folder_path = '/path/to/folder'
filenames = get_all_filenames(folder_path)
print('All Filenames:')
for filename in filenames:
    print(filename)

pattern = r'\.txt$'
filtered_filenames = filter_filenames(filenames, pattern)
print('\nFiltered Filenames:')
for filename in filtered_filenames:
    print(filename)

file_types = count_file_types(filenames)
plot_pie_chart(file_types)

save_results(filenames, filtered_filenames, file_types)
print('Results saved to results.txt.')