os
模块是Python中用于与操作系统进行交互的标准库,提供了许多操作文件和目录、进程管理等功能,以下是其主要用法:
文件和目录操作
- 获取当前工作目录:使用
os.getcwd()
函数可以获取当前Python脚本所在的工作目录。示例代码如下:
import os
current_dir = os.getcwd()
print(current_dir)
- 改变当前工作目录:
os.chdir(path)
函数用于改变当前工作目录到指定的路径path
。示例代码如下:
import os
new_dir = "C:/Users/Admin/Documents"
os.chdir(new_dir)
current_dir = os.getcwd()
print(current_dir)
- 创建目录:
os.mkdir(path)
函数用于创建一个新的目录。如果要创建多级目录,可以使用os.makedirs(path)
函数。示例代码如下:
import os
new_dir = "C:/Users/Admin/Documents/new_folder"
os.mkdir(new_dir)
new_multi_dir = "C:/Users/Admin/Documents/multi_folder/sub_folder"
os.makedirs(new_multi_dir)
- 删除目录:
os.rmdir(path)
函数用于删除空目录。如果要删除非空目录及其所有内容,需要使用shutil
模块的rmtree
函数。示例代码如下:
import os
import shutil
empty_dir = "C:/Users/Admin/Documents/empty_folder"
os.rmdir(empty_dir)
non_empty_dir = "C:/Users/Admin/Documents/non_empty_folder"
shutil.rmtree(non_empty_dir)
- 列出目录内容:
os.listdir(path)
函数返回指定目录下的所有文件和目录的名称列表。示例代码如下:
import os
dir_content = os.listdir("C:/Users/Admin/Documents")
print(dir_content)
- 文件重命名:
os.rename(src, dst)
函数用于将文件或目录从源路径src
重命名为目标路径dst
。示例代码如下:
import os
old_name = "C:/Users/Admin/Documents/old_file.txt"
new_name = "C:/Users/Admin/Documents/new_file.txt"
os.rename(old_name, new_name)
- 删除文件:
os.remove(path)
函数用于删除指定路径的文件。示例代码如下:
import os
file_path = "C:/Users/Admin/Documents/delete_file.txt"
os.remove(file_path)
进程管理
- 获取进程ID:使用
os.getpid()
函数可以获取当前进程的ID。示例代码如下:
import os
pid = os.getpid()
print(pid)
- 获取父进程ID:在Unix和类Unix系统中,可以使用
os.getppid()
函数获取当前进程的父进程ID。示例代码如下:
import os
ppid = os.getppid()
print(ppid)
- 执行系统命令:
os.system(command)
函数可以在Python中执行系统命令,并返回命令的返回值。示例代码如下:
import os
command = "dir"
os.system(command)
环境变量操作
- 获取环境变量:
os.environ
是一个包含所有环境变量的字典,可以通过键来获取特定的环境变量值。示例代码如下:
import os
path = os.environ.get("PATH")
print(path)
- 设置环境变量:使用
os.environ[key] = value
的方式可以设置环境变量。示例代码如下:
import os
os.environ["MY_VAR"] = "my_value"
print(os.environ.get("MY_VAR"))