如何在Python中设置当前工作目录？

在Python项目中，很多时候需要在代码中引用其他文件或者数据，而这些文件或数据通常会存放在不同的目录中。设置当前工作目录可以帮助我们在代码中更方便地访问这些文件或数据。本文将介绍如何在Python中设置当前工作目录。

获取当前工作目录

在Python中，我们可以调用os.getcwd()函数获取当前工作目录（Current Working Directory，简称cwd）的路径。下面是一个示例代码：

import os
print(os.getcwd())

代码输出结果：

/home/user/projects/myproject

注意事项：
– 在不同的操作系统上，当前工作目录的表达方式可能会有所不同。
– 建议在代码中使用相对路径来访问其他文件或数据，更加灵活和可移植。

设置当前工作目录

如果我们需要更改当前工作目录，可以调用os.chdir(path)函数，将当前工作目录改为指定的路径path。下面是一个示例代码：

import os
new_path = "/home/user/myfiles"
os.chdir(new_path)
print(os.getcwd())

代码输出结果：

/home/user/myfiles

注意事项：
– 设置当前工作目录时，建议使用绝对路径来避免路径拼接错误等问题。
– Python在退出程序时会自动将当前工作目录改回程序运行前的状态，不需要手动改回。

示例：读取其他目录下的文件

假设我们的项目目录结构如下：

myproject/
├── data/
│   ├── file1.txt
│   └── file2.txt
└── src/
    ├── main.py
    └── helper.py

我们在helper.py中编写了一个函数，用于读取data目录下的file1.txt文件：

import os

def read_file():
    data_dir = "../data"
    file_path = os.path.join(data_dir, "file1.txt")
    with open(file_path, "r") as f:
        data = f.read()
    return data

在代码中，我们使用了相对路径和os.path.join()函数来构造file1.txt的完整路径。假设我们需要在main.py中调用read_file()函数，那么我们需要在main.py中首先将当前工作目录修改为项目根目录，然后再调用read_file()函数：

import os
from src.helper import read_file

project_dir = os.path.abspath(os.path.join(os.getcwd(), ".."))
os.chdir(project_dir)

data = read_file()
print(data)

在代码中，我们首先构造了项目根目录的路径，然后将当前工作目录修改为该路径。接着调用read_file()函数来读取file1.txt文件，并将其打印出来。如果我们不将当前工作目录修改为项目根目录，将无法正确访问../data/file1.txt文件。