Python遍历文件夹中的文件并根据文件类型将文件分类储存

合集 - 一些代码(14)

1.MXnet深度学习环境及GluonCV深度学习包的安装2023-07-26

2.Python遍历文件夹中的文件并根据文件类型将文件分类储存2023-09-06

3.对ERA5数据进行区域提取及时间处理2023-09-134.基于街景全景照片数据计算SVF（天空开阔度）2023-09-225.基于transbigdata包获取交通线路和站点的邻接表及shp文件2023-09-226.gephi导入networkx：使用经纬度绘图并根据情景计算节点指标与网络整体指标（关联gephi导入networkx一文）2023-09-227.备忘：爬取poi代码2023-09-238.基于Python+MXnet预训练模型的街景图像语义分割代码2023-11-079.基于Python Sklearn的机器学习代码（备忘）2023-12-0910.基于OpenAI API接口向GPT4v上传图像进行图像分析2024-06-0211.保持校园网自动登录的代码备忘: 基于webdriver和chrome2024-07-0312.Python Selenium+cookie+XPATH爬取数据2024-07-1013.Python web自动化爬虫-selenium/处理验证码/Xpath2024-07-1814.0803/代码2024-08-03

收起

最近在写某串代码，其逻辑是从源文件夹中遍历各个子文件夹，将其中不同文件类型的文件分类提取出来各自使用，需要提取.txt、.png、.csv文件三种，为了实现这一功能，用os包来实现

import os

#准备文件路径
for n in range(0,12854):    #这里我的文件名是0-12854，如果文件名不是循环的话需要再os.walk一下获取子文件夹
    print('The No.{} coordinate is handling'.format(n))
    general_path='路径/{}'.format(n)
    try:
        general_file=os.walk(general_path)
    except FileNotFoundError:
        print("No.{} coordinate is missing".format(n))

#创建保存文件名称的列表
    img_names=[]
    lonlat_names=[]
    csv_names=[]

#将各文件的名字提取到列表中 
    for root, dirs, files in general_file:
        img_names[:] = [f for f in files if f.endswith(".png")]
        lonlat_names[:] = [g for g in files if g.endswith(".txt")]
        csv_names[:] = [h for h in files if h.endswith(".csv")]

#从列表中遍历某类型的文件
        for lonlat in lonlat_names:
            print(lonlat)