[python] 디렉토리에서 필터링 된 파일 목록 가져 오기

Python을 사용하여 디렉토리에 파일 목록을 가져 오려고하지만 모든 파일 목록을 원하지 않습니다.

내가 본질적으로 원하는 것은 다음과 같지만 파이썬을 사용하고 ls를 실행하지 않는 기능을 수행하는 능력입니다.

ls 145592*.jpg

내장 된 방법이 없다면 현재 for 루프를 작성하여 결과를 반복 os.listdir()하고 일치하는 모든 파일을 새 목록에 추가하려고합니다.

그러나 해당 디렉토리에는 많은 파일이 있으므로 더 효율적인 방법 (또는 내장 방법)이 있기를 바랍니다.



답변


답변

glob.glob()(Ignacio에 따라) 확실히하는 방법입니다. 그러나 더 복잡한 일치가 필요한 경우 목록 이해 및 re.match()로 다음과 같이 할 수 있습니다.

files = [f for f in os.listdir('.') if re.match(r'[0-9]+.*\.jpg', f)]

보다 유연하지만, 효율적이지 않습니다.


답변

간단하게 유지하십시오.

import os
relevant_path = "[path to folder]"
included_extensions = ['jpg','jpeg', 'bmp', 'png', 'gif']
file_names = [fn for fn in os.listdir(relevant_path)
              if any(fn.endswith(ext) for ext in included_extensions)]

영어로 잘 읽히기 때문에 이런 형태의 목록 이해를 선호합니다.

네 번째 줄은 다음과 같이 읽습니다. 내 경로에 대해 os.listdir의 각 fn에 대해 포함 된 확장 중 하나와 일치하는 것을 제공하십시오.

초보자 파이썬 프로그래머는 필터링에 목록 이해를 사용하는 데 실제로 익숙하지 않을 수 있으며 매우 큰 데이터 세트에 대해 약간의 메모리 오버 헤드가있을 수 있지만 디렉토리 및 기타 간단한 문자열 필터링 작업을 나열하는 경우 목록 이해가 더 깨끗합니다. 문서화 가능한 코드.

이 디자인의 유일한 점은 목록 대신 문자열을 전달하는 실수를 방지하는 것입니다. 예를 들어 실수로 문자열을 목록으로 변환하고 문자열의 모든 문자를 검사하면 잘못된 긍정 오류가 발생할 수 있습니다.

그러나 이해하기 어려운 솔루션보다 수정하기 쉬운 문제가있는 것이 좋습니다.


답변

다른 옵션 :

>>> import os, fnmatch
>>> fnmatch.filter(os.listdir('.'), '*.py')
['manage.py']

https://docs.python.org/3/library/fnmatch.html


답변

glob모듈이있는 필터 :

글로브 가져 오기

import glob

와일드 카드 :

files=glob.glob("data/*")
print(files)

Out:

['data/ks_10000_0', 'data/ks_1000_0', 'data/ks_100_0', 'data/ks_100_1',
'data/ks_100_2', 'data/ks_106_0', 'data/ks_19_0', 'data/ks_200_0', 'data/ks_200_1',
'data/ks_300_0', 'data/ks_30_0', 'data/ks_400_0', 'data/ks_40_0', 'data/ks_45_0',
'data/ks_4_0', 'data/ks_500_0', 'data/ks_50_0', 'data/ks_50_1', 'data/ks_60_0',
'data/ks_82_0', 'data/ks_lecture_dp_1', 'data/ks_lecture_dp_2']

Fiter 확장 .txt:

files = glob.glob("/home/ach/*/*.txt")

단일 문자

glob.glob("/home/ach/file?.txt")

숫자 범위

glob.glob("/home/ach/*[0-9]*")

알파벳 범위

glob.glob("/home/ach/[a-c]*")


답변

예비 코드

import glob
import fnmatch
import pathlib
import os

pattern = '*.py'
path = '.'

해결 방법 1- “글로브”사용

# lookup in current dir
glob.glob(pattern)

In [2]: glob.glob(pattern)
Out[2]: ['wsgi.py', 'manage.py', 'tasks.py']

해결 방법 2- “os”+ “fnmatch”사용

변형 2.1- 현재 디렉토리의 조회

# lookup in current dir
fnmatch.filter(os.listdir(path), pattern)

In [3]: fnmatch.filter(os.listdir(path), pattern)
Out[3]: ['wsgi.py', 'manage.py', 'tasks.py']

변형 2.2- 조회 재귀

# lookup recursive
for dirpath, dirnames, filenames in os.walk(path):

    if not filenames:
        continue

    pythonic_files = fnmatch.filter(filenames, pattern)
    if pythonic_files:
        for file in pythonic_files:
            print('{}/{}'.format(dirpath, file))

결과

./wsgi.py
./manage.py
./tasks.py
./temp/temp.py
./apps/diaries/urls.py
./apps/diaries/signals.py
./apps/diaries/actions.py
./apps/diaries/querysets.py
./apps/library/tests/test_forms.py
./apps/library/migrations/0001_initial.py
./apps/polls/views.py
./apps/polls/formsets.py
./apps/polls/reports.py
./apps/polls/admin.py

해결 방법 3- “pathlib”사용

# lookup in current dir
path_ = pathlib.Path('.')
tuple(path_.glob(pattern))

# lookup recursive
tuple(path_.rglob(pattern))

노트:

  1. 파이썬 3.4에서 테스트
  2. “pathlib”모듈은 Python 3.4에서만 추가되었습니다
  3. Python 3.5에는 glob.glob https://docs.python.org/3.5/library/glob.html#glob.glob 를 사용한 재귀 조회 기능이 추가되었습니다
    . 내 컴퓨터는 Python 3.4와 함께 설치되었으므로 테스트하지 않았습니다.

답변

os.walk를 사용하여 파일을 재귀 적으로 나열하십시오.

import os
root = "/home"
pattern = "145992"
alist_filter = ['jpg','bmp','png','gif']
path=os.path.join(root,"mydir_to_scan")
for r,d,f in os.walk(path):
    for file in f:
        if file[-3:] in alist_filter and pattern in file:
            print os.path.join(root,file)