집 > 기사 > 백엔드 개발 > Python3. 특정 파일 이름을 추출하기 위해 폴더를 탐색하는 예

Python3. 특정 파일 이름을 추출하기 위해 폴더를 탐색하는 예

不言원래의: 2018-04-27 10:49:032323검색

다음은 Python 3에서 특정 파일명을 추출하기 위해 폴더를 순회하는 예제입니다. 좋은 참고값이 있어서 모두에게 도움이 되었으면 좋겠습니다. 같이 살펴보시죠

파일을 일괄 처리하다 보면 특정 조건에서 파일명을 추출하기 위해 특정 경로를 순회해야 하는 경우가 종종 있습니다. 이 글은 폭력적인 순회이지만 매우 간결한 방법을 작성합니다. 정말 매우 간결하지만 매우 폭력적입니다.

예제의 목표는 다음과 같습니다. 원격탐사 데이터가 저장된 폴더 아래에 이름이 "_BAD"로 끝나는 폴더의 내용을 가져옵니다. 이 파일 아래에는 여러 레벨(년/월/제품 유형/)이 있기 때문에 대상 폴더도 많고 마지막 레벨에 존재하기 때문에 수동으로 확인하기가 매우 귀찮습니다.

코드는 다음과 같습니다(지식 포인트를 요약한 후):

# -*- coding: utf-8 -*-
"""
遍历某路径下所有文件夹，获得特定文件夹下所有文件
很暴力，真的遍历了所有的文件夹
20180124
@author: 墨大宝
"""
import os
TARGETPATH = r&#39;F:\MODIS_DATA&#39;
records = []
for currentDir, _, includedFiles in os.walk(TARGETPATH):
 if not currentDir.endswith(&#39;_BAD&#39;): continue
 else:
  records.append(currentDir) # 将以“_BAD”结尾的文件夹名加入records
  records.extend(includedFiles) # 将该文件夹内的文件名列表扩展到records
# 将records写入.txt
txtFile = open(os.path.join(TARGETPATH, &#39;02_04_BAD.txt&#39;), &#39;w&#39;)
txtFile.write(os.linesep.join(records))
txtFile.close()
# 将排序后的records写入.txt
with open(os.path.join(TARGETPATH, &#39;02_04_BAD_SORTED.txt&#39;), &#39;w&#39;) as txtFile:
 txtFile.write(&#39;\n&#39;.join(sorted(records)))

os.walk()는 디렉토리 트리 생성기를 반환합니다. (dirpath, dirnames, filenames) 형식의 튜플이 생성될 때마다 요소는 현재 경로, 현재 경로 아래의 폴더 목록, 현재 경로 아래의 파일 이름 목록입니다.

List의 .append(), .extend() 및 .sort() 메서드는 모두 제자리에서 수정되지만 sorted() 함수는 그렇지 않습니다.

목록을 .txt 파일로 작성할 때 목록을 str로 변환해야 합니다. str() 함수를 직접 사용하여 변환을 강제하는 것은 보기에 좋지 않습니다.

os.path는 시스템 개행 문자를 나타내며 Windows에서는 "rn"이고 다른 시스템에서는 "n"입니다. 그러나 os.path를 사용하거나 "n"을 사용하여 목록 요소를 연결하고 마지막으로 Windows 메모장으로 열면 줄 바꿈이 동일하지만 vs 코드로 열면 os.path가 변경됩니다. 한 줄이 더 많아지면 줄 사이에 한 줄처럼 보일 것입니다. 그 이유는 무엇입니까? 말하자면, Python의 쓰기 메커니즘과 관련이 있을 수 있으므로 당분간은 자세히 다루지 않겠습니다. ).

파일 읽기 및 쓰기와 관련하여 대부분의 정보에서는 with 형식을 권장하는데 이는 실제로 더 간결합니다.

PS:

os.walk()는 디렉토리 트리에 따라 주어진 경로의 모든 폴더와 파일을 실제로 순회하기 때문에 폭력적이라고 합니다. 파일의 양은 많지만 파일 이름은 많습니다. 속도가 느려질 것입니다(실제로는 그다지 느리지 않다고 생각합니다). os.listdir()을 사용하여 재귀 함수를 작성하면 실행 효율성이 더 높아질 수 있지만 논리는 다음과 같습니다. os.walk()는 간단하고 작성하기 쉽습니다. 자유롭게 해보세요. 제가 해냈습니다!

Python3. 특정 파일 이름을 추출하기 위해 폴더를 탐색하는 예

관련 기사