Python 同级目录下的文件及其操作

在Python编程中,我们经常需要操作文件来读取、写入和处理数据。而文件通常存储在同级目录中,也就是与我们的Python脚本位于同一个文件夹下。本文将介绍如何在Python中操作同级目录下的文件,并提供相应的代码示例。

1. 查看和改变当前工作目录

在Python中,我们可以使用os模块来操作文件和目录。首先,我们需要了解如何查看和改变当前的工作目录。

import os

# 获取当前工作目录
current_dir = os.getcwd()
print("当前工作目录:", current_dir)

# 改变当前工作目录
os.chdir("path/to/new/directory")

在上面的代码中,os.getcwd()函数用于获取当前工作目录并将其存储在变量current_dir中。os.chdir()函数用于改变当前工作目录到指定的路径。

2. 列出目录中的文件

一旦我们知道当前的工作目录,我们可以使用os.listdir()函数来获取目录中的所有文件和文件夹。

import os

# 列出当前工作目录中的文件和文件夹
files = os.listdir()
for file in files:
    print(file)

上述代码将列出当前工作目录中的所有文件和文件夹,并逐行打印出来。

3. 检查文件是否存在

在操作文件之前,我们通常需要检查文件是否存在。我们可以使用os.path.exists()函数来检查文件是否存在。

import os

# 检查文件是否存在
file_path = "example.txt"
if os.path.exists(file_path):
    print("文件存在")
else:
    print("文件不存在")

上述代码中,我们将文件路径存储在file_path变量中,然后使用os.path.exists()函数来检查文件是否存在。如果文件存在,将打印"文件存在";否则,将打印"文件不存在"。

4. 打开和关闭文件

接下来,我们将学习如何打开和关闭文件。在Python中,我们可以使用open()函数来打开一个文件,并使用close()方法来关闭文件。

# 打开文件
file = open("example.txt", "r")

# 读取文件内容
content = file.read()
print(content)

# 关闭文件
file.close()

上述代码中,我们使用open()函数打开一个名为"example.txt"的文件,并将其存储在file变量中。然后,使用read()方法读取文件的内容,并将其存储在content变量中。最后,使用close()方法关闭文件。

5. 写入文件

在Python中,我们可以使用write()方法将数据写入文件。

# 打开文件
file = open("example.txt", "w")

# 写入文件
file.write("Hello, world!")

# 关闭文件
file.close()

上述代码中,我们使用open()函数将文件"example.txt"以写入模式打开,并将其存储在file变量中。然后,使用write()方法将字符串"Hello, world!"写入文件中。最后,使用close()方法关闭文件。

6. 删除文件

如果我们想删除同级目录下的某个文件,可以使用os.remove()函数。

import os

# 删除文件
file_path = "example.txt"
if os.path.exists(file_path):
    os.remove(file_path)
    print("文件已删除")
else:
    print("文件不存在")

上述代码中,我们使用os.remove()函数来删除文件。首先,我们通过os.path.exists()函数检查文件是否存在。如果文件存在,将使用os.remove()函数删除文件并打印"文件已删除";否则,将打印"文件不存在"。

序列图

以下是一个使用上述代码操作同级目录下文件的序列图:

sequenceDiagram
    participant User
    participant Python Script
    participant File System

    User->>Python Script: 执行Python脚本
    Python Script->>File System: 获取当前工作目录
    File System->>Python Script: 返回当前工作目录
    Python Script->>File System: 列出目录中的文件
    File System->>Python Script: 返回