Python 中的 os 模块 - 目录操作

Python 中的 os 模块提供了一些用于操作目录的函数。有关 os 模块的更多信息，请参阅Python Docs。

获取当前目录

可以使用 os.getcwd() 函数获取当前目录的绝对路径。

import os

current_directory = os.getcwd()
print("当前目录：", current_directory)

输出：

当前目录： /Users/username/Documents

切换目录

使用 os.chdir() 函数可以更改当前工作目录。

import os

os.chdir('/Users/username/Documents/code')
print("当前目录：", os.getcwd())

输出：

当前目录： /Users/username/Documents/code

列出目录中的文件

使用 os.listdir() 函数列出指定目录中的所有文件和目录。

import os

files = os.listdir('/Users/username/Documents/code')
print("文件列表：", files)

输出：

文件列表： ['file1.txt', 'file2.pdf', 'subdirectory', 'test.py']

创建目录

os.mkdir() 函数可以在指定路径下创建一个新目录。

import os

os.mkdir('/Users/username/Documents/code/new_directory')

删除目录

os.rmdir() 函数可以删除指定的空目录。

import os

os.rmdir('/Users/username/Documents/code/new_directory')

删除文件

使用 os.remove() 函数可以删除指定的文件。

import os

os.remove('/Users/username/Documents/code/file1.txt')

检查目录或文件是否存在

使用 os.path.exists() 函数可以检查指定的路径是否存在。该函数将返回 True 或 False。

import os

if os.path.exists('/Users/username/Documents/code/file1.txt'):
    print("文件存在！")
else:
    print("文件不存在。")

输出：

文件存在！

获取目录中的所有文件

可以使用 os.walk() 函数来获取指定目录中的所有文件。

import os

folder_path = '/Users/username/Documents/code'
for dirpath, dirnames, filenames in os.walk(folder_path):
    print(f'--- {dirpath} ---')
    print(f'\tDirectories: {dirnames}')
    print(f'\tFiles: {filenames}')

输出：

--- /Users/username/Documents/code ---
    Directories: ['subdirectory']
    Files: ['file1.txt', 'file2.pdf', 'test.py']
--- /Users/username/Documents/code/subdirectory ---
    Directories: []
    Files: ['subfile.txt']

总结

在 Python 中，os 模块提供了丰富的功能，使我们可以轻松地操作文件和目录。本文介绍了获取当前目录、切换目录、列出目录中的文件、创建目录、删除目录和文件、检查目录或文件是否存在以及获取目录中的所有文件的方法。