Comment calculer la taille du répertoire en Python ?

Calcul de la taille d'un répertoire avec Python

Le calcul de la taille d'un répertoire en Python peut être une tâche utile pour gérer l'espace de stockage ou analyser des données. Explorons comment calculer efficacement cette taille.

Sumer les tailles de fichiers à l'aide de os.walk

Une approche consiste à parcourir le répertoire et ses sous-répertoires, en additionnant les tailles de fichiers. Cela peut être implémenté à l'aide de la fonction os.walk :

<code class="python">import os

def get_size(start_path='.'):
    total_size = 0
    for dirpath, dirnames, filenames in os.walk(start_path):
        for f in filenames:
            fp = os.path.join(dirpath, f)
            if not os.path.islink(fp):
                total_size += os.path.getsize(fp)
    return total_size

print(get_size(), 'bytes')</code>

Cette fonction calcule de manière récursive la taille du répertoire, fournissant la taille totale en octets.

One-liner Using os. listdir

Pour un calcul rapide de la taille du répertoire sans tenir compte des sous-répertoires, vous pouvez utiliser une seule ligne :

<code class="python">import os
sum(os.path.getsize(f) for f in os.listdir('.') if os.path.isfile(f))</code>

Cette expression utilise os.listdir pour lister tous les fichiers dans le répertoire actuel et os.path.getsize pour déterminer leurs tailles.

En utilisant os.stat et os.scandir

Vous pouvez également utiliser os.stat ou os.scandir pour calculer la taille des fichiers. os.stat fournit des informations supplémentaires sur les fichiers, notamment leur taille :

<code class="python">import os

nbytes = sum(d.stat().st_size for d in os.scandir('.') if d.is_file())</code>

os.scandir offre des performances améliorées dans Python 3.5 et offre un moyen plus efficace de parcourir les répertoires.

Pathlib Solution

Si vous utilisez Python 3.4, la bibliothèque pathlib offre un moyen pratique de gérer les opérations de répertoire :

<code class="python">from pathlib import Path

root_directory = Path('.')
sum(f.stat().st_size for f in root_directory.glob('**/*') if f.is_file())</code>

Cette solution pathlib combine les approches précédentes pour un calcul efficace.

Ce qui précède est le contenu détaillé de. pour plus d'informations, suivez d'autres articles connexes sur le site Web de PHP en chinois!