Home > Article > Backend Development > How to Calculate Directory Size in Python?
Calculating Directory Size with Python
Calculating the size of a directory in Python can be a useful task for managing storage space or analyzing data. Let's explore how to efficiently calculate this size.
Sum File Sizes Using os.walk
One approach involves traversing the directory and its subdirectories, summing the file sizes. This can be implemented using the os.walk function:
<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>
This function recursively computes the directory size, providing the total size in bytes.
One-liner Using os.listdir
For a quick calculation of the directory size without considering subdirectories, you can use a one-liner:
<code class="python">import os sum(os.path.getsize(f) for f in os.listdir('.') if os.path.isfile(f))</code>
This expression employs os.listdir to list all files in the current directory, and os.path.getsize to determine their sizes.
Using os.stat and os.scandir
Alternatively, you can use os.stat or os.scandir for calculating file sizes. os.stat provides additional file information, including size:
<code class="python">import os nbytes = sum(d.stat().st_size for d in os.scandir('.') if d.is_file())</code>
os.scandir offers improved performance in Python 3.5 and offers a more efficient way to iterate through directories.
Pathlib Solution
If you're using Python 3.4 , the pathlib library offers a convenient way to handle directory operations:
<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>
This pathlib solution combines the previous approaches for a concise and efficient calculation.
The above is the detailed content of How to Calculate Directory Size in Python?. For more information, please follow other related articles on the PHP Chinese website!