Python读取指定文件夹下指定类型数据的文件名并保存到TXT文件中

最新推荐文章于 2023-01-16 16:33:22 发布

原创

最新推荐文章于 2023-01-16 16:33:22 发布 · 1.2k 阅读

2 ·

CC 4.0 BY-SA版权

文章标签：

#python

一、读取路径到指定TXT文件

有31个DEM文件夹，每个文件夹下有3个文件：ASTGTM2_N34E115_dem.tif，ASTGTM2_N34E115_num.tif，README.tif。想读取每个DEM文件夹下的ASTGTM2_N34E115_dem.tif文件。

代码如下：

import os

# 查询符合条件的文件
f = open(r'D:\ProfessionalProfile\VectorRelevant\GDEMV2_30m_shandong\DEM1.txt','w')
j,k=0,0
def file(root, ext):

    for i in os.listdir(root):  # os.listdir(root) 读取root目录下的文件和目录

        # os.path.splitext()将文件名和扩展名分开
        ext = os.path.splitext(root + '\\' + i)  # os.path.splitext()读取文件扩展名

        if ext[1]:  # 有扩展名时 为文件
            global j
            if i.count(ex) > 0:  # 包含特定字符
                pathDEM = root +'\\'+ i
                f.write(pathDEM + '\n')
                f.flush()
                # print(pathDEM)  # 输出