Python 更改当前工作目录|极客笔记

Python 更改当前工作目录

在Python中，我们经常需要在不同的目录下读取或写入文件。为了方便操作这些文件，我们需要先更改当前工作目录。在本文中，我将详细介绍如何在Python中更改当前工作目录。

1. 使用os模块更改工作目录

Python内置的os模块提供了一种简单的方法来更改当前工作目录。我们可以使用os.chdir()函数来实现在Python中更改当前的工作目录。

import os

# 打印当前工作目录
print("当前工作目录:", os.getcwd())

# 更改工作目录为指定目录
os.chdir("/path/to/directory")

# 打印更改后的工作目录
print("更改后的工作目录:", os.getcwd())

在上面的示例中，首先我们通过os.getcwd()函数打印了当前的工作目录，然后使用os.chdir()函数将工作目录更改为指定的目录。最后再次打印工作目录，确认是否已经更改。

2. 使用contextlib和os模块临时更改工作目录

有时候我们只想暂时更改工作目录，完成某些操作后又要恢复原来的工作目录。在这种情况下，可以使用contextlib模块中的contextmanager和os模块来实现暂时更改工作目录的功能。

import os
from contextlib import contextmanager

@contextmanager
def change_dir(new_dir):
    old_dir = os.getcwd()
    os.chdir(new_dir)
    try:
        yield
    finally:
        os.chdir(old_dir)


# 打印当前工作目录
print("当前工作目录:", os.getcwd())

# 使用with语句临时更改工作目录
with change_dir("/path/to/directory"):
    # 在新的工作目录下执行操作
    print("当前工作目录(暂时更改):", os.getcwd())

# 打印恢复后的工作目录
print("原始工作目录:", os.getcwd())

在上面的示例中，我们定义了一个change_dir装饰器，利用contextmanager来创建一个上下文管理器。在with语句块中，我们可以临时更改工作目录并进行相应操作，而在with语句块结束后，工作目录会自动恢复到之前的状态。

3. 运行结果

下面是上述示例代码的运行结果：

当前工作目录: /home/user
更改后的工作目录: /path/to/directory
当前工作目录(暂时更改): /path/to/directory
原始工作目录: /home/user

通过以上示例，我们学会了在Python中如何更改当前的工作目录。无论是永久性更改或者临时更改，我们都可以轻松地实现这一功能。

Python 更改当前工作目录