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"))