Python 当前目录下查找文件

在编程过程中，有时候我们需要在当前目录下查找特定的文件。Python 提供了一些方法来实现这个功能。在本文中，我们将介绍如何使用 Python 在当前目录下查找文件，并提供了相关的代码示例。

当前目录

在开始之前，我们需要了解什么是当前目录。当前目录是指正在运行的 Python 脚本所在的目录。可以通过 os 模块的 getcwd() 函数来获取当前目录的路径。

import os

current_dir = os.getcwd()
print("当前目录：", current_dir)

以上代码将打印出当前运行脚本所在的目录。

查找文件

一旦我们知道了当前目录，就可以使用相关的函数来查找文件了。Python 提供了 os 模块和 glob 模块来实现这个功能。

使用 os 模块

os 模块提供了一些方法用于操作文件和目录。其中，os.listdir() 函数可以获取当前目录下的所有文件和目录的名称列表。

import os

files = os.listdir()
print("当前目录下的文件和目录：", files)

以上代码将打印出当前目录下的所有文件和目录的名称列表。

如果我们只想查找特定类型的文件，可以使用 os.path 模块的 isfile() 函数，结合列表推导式来过滤文件。

import os

files = [file for file in os.listdir() if os.path.isfile(file)]
print("当前目录下的文件：", files)

以上代码将打印出当前目录下的所有文件的名称列表。

使用 glob 模块

glob 模块提供了一个函数用于读取符合特定模式的文件列表。可以使用通配符来匹配文件名。

import glob

files = glob.glob("*")
print("当前目录下的文件和目录：", files)

以上代码将打印出当前目录下的所有文件和目录的名称列表。

如果我们只想查找特定类型的文件，可以使用通配符来匹配文件类型。

import glob

files = glob.glob("*.txt")
print("当前目录下的文本文件：", files)

以上代码将打印出当前目录下所有的文本文件的名称列表。

代码示例

下面是一个完整的示例，演示了如何使用 Python 在当前目录下查找文件。

import os

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

# 使用 os 模块查找文件
files = [file for file in os.listdir() if os.path.isfile(file)]
print("当前目录下的文件：", files)

# 使用 glob 模块查找文件
files = glob.glob("*")
print("当前目录下的文件和目录：", files)

# 使用 glob 模块查找特定类型的文件
files = glob.glob("*.txt")
print("当前目录下的文本文件：", files)

关系图

下面是一个关系图，描述了在当前目录下查找文件的过程。

erDiagram
      os模块 }|--| 获取当前目录
      os模块 }|--| 查找文件
      glob模块 }|--| 查找文件

类图

下面是一个类图，展示了相关的类和函数。

classDiagram
      os模块 <|-- os
      glob模块 <|-- glob
      os模块 : getcwd()
      os模块 : listdir()
      os模块 : path
      glob模块 : glob()

在本文中，我们介绍了如何使用 Python 在当前目录下查找文件。我们学习了如何获取当前目录的路径，以及如何使用 os 模块和 glob 模块来查找文件。希望本文能对你有所帮助！