Qual è il modo Python di camminare su un albero delle directory?

Question

Ritengo che assegnare file e cartelle e fare la parte + = [item] sia un po 'hackerato. Eventuali suggerimenti? Sto usando Python 3,2

from os import * 
from os.path import * 

def dir_contents(path): 
    contents = listdir(path) 
    files = [] 
    folders = [] 
    for i, item in enumerate(contents): 
     if isfile(contents[i]): 
      files += [item] 
     elif isdir(contents[i]): 
      folders += [item] 
    return files, folders 

Answer 1

Date un'occhiata alla funzione os.walk che restituisce il percorso lungo con le directory ei file in essa contenuti. Ciò dovrebbe ridurre notevolmente la tua soluzione.

Answer 2

Provare a utilizzare il metodo append.

Answer 3

def dir_contents(path): 
    files,folders = [],[] 
    for p in listdir(path): 
     if isfile(p): files.append(p) 
     else: folders.append(p) 
    return files, folders 

Answer 4

Infatti utilizzando

items += [item] 

è male per molte ragioni ...

1. Procedimento append è stata fatta esattamente per questo (aggiungendo un elemento al fine una lista)

2. Tu stanno creando un elenco temporaneo di un elemento solo per buttarlo via. Mentre la velocità grezza non dovrebbe essere la tua prima preoccupazione quando usi Python (altrimenti stai utilizzando la lingua sbagliata) la velocità sprecata senza motivo non sembra la cosa giusta.

3. Si sta utilizzando un po 'asimmetria del linguaggio Python ... per oggetti lista scrittura a += b non è lo stesso che scrivere a = a + b perché il primo modifica l'oggetto sul posto, mentre il secondo, invece alloca una nuova lista e questo può avere una semantica diversa se l'oggetto a è raggiungibile anche in altri modi. Nel tuo codice specifico questo non sembra il caso, ma potrebbe diventare un problema più tardi quando qualcun altro (o te stesso in pochi anni, è lo stesso) dovrà modificare il codice. Python ha anche un metodo extend con una sintassi meno sottile specificatamente creata per gestire il caso in cui si desidera modificare un oggetto elenco aggiungendo alla fine gli elementi di un altro elenco.

anche come altri hanno notato sembra che il codice sta cercando di fare quello che fa già os.walk ...

Answer 5

Invece del built-in os.walk e os.path.walk, io uso qualcosa derivato da questo pezzo di codice che ho trovato suggerito altrove:

http://code.google.com/p/mylibs/source/browse/lib/Python/MyPyLib/DirectoryStatWalker.py

non voglio repaste qui, ma cammina la directory in modo ricorsivo ed è abbastanza efficiente e di facile lettura.

Answer 6

Mentre ho cercato su Google le stesse informazioni, ho trovato questa domanda.

Sto postando qui il codice più piccolo e più chiaro che ho trovato in http://www.pythoncentral.io/how-to-traverse-a-directory-tree-in-python-guide-to-os-walk/ (piuttosto che semplicemente postando l'URL, in caso di link rot).

La pagina contiene alcune informazioni utili e indica anche altre pagine pertinenti.

# Import the os module, for the os.walk function 
import os 

# Set the directory you want to start from 
rootDir = '.' 
for dirName, subdirList, fileList in os.walk(rootDir): 
    print('Found directory: %s' % dirName) 
    for fname in fileList: 
     print('\t%s' % fname) 

Answer 7

non ho ancora testato estensivamente questo, ma credo che questo amplierà il generatore os.walk, unire dirnames a tutti i percorsi dei file, e appiattire la lista risultante; Per dare un elenco diretto di file concreti nel percorso di ricerca.

import itertools 
import os 

def find(input_path): 
    return itertools.chain(
     *list(
      list(os.path.join(dirname, fname) for fname in files) 
      for dirname, _, files in os.walk(input_path) 
     ) 
    ) 

Answer 8

Se volete iterare ricorsivamente tutti i file, compresi tutti i file nelle sottocartelle, credo che questo sia il modo migliore.

import os 

def get_files(input): 
    for fd, subfds, fns in os.walk(input): 
     for fn in fns: 
      yield os.path.join(fd, fn) 

## now this will print all full paths 

for fn in get_files(fd): 
    print(fn) 

Answer 9

Da Python 3.4 c'è un nuovo modulo pathlib. Quindi per ottenere tutti i file e le dir si può fare:

from pathlib import Path 

dirs = [str(item) for item in Path(path).iterdir() if item.is_dir()] 
files = [str(item) for item in Path(path).iterdir() if item.is_file()]