极客笔记Python 3 – os.listdir() 方法
在Python 3中，os.listdir()方法可以列出指定目录中的文件和文件夹。这是一种十分有用的方法，尤其在处理文件和文件夹时。本文将介绍os.listdir()方法的使用和一些注意事项。
基本语法
os.listdir()方法的基本语法如下：
os.listdir(path)
其中，path是必须的参数，用于指定要列出内容的目录路径。
示例
下面是os.listdir()方法的一些示例代码：
# 列出当前目录下的所有文件和文件夹
import os
current_dir = os.getcwd()
print(os.listdir(current_dir))
上面的代码将输出当前目录下的所有文件和文件夹列表。如果你运行上面的代码，你将会看到一个输出，其中包括当前目录下的所有文件和文件夹的名字。
现在，让我们看看如何使用os.listdir()方法来过滤出指定类型的文件。例如，如果我们想要列出当前目录下的所有Python文件，我们可以这样写：
# 列出当前目录下的所有.py文件
import os
current_dir = os.getcwd()
for file in os.listdir(current_dir):
if file.endswith(".py"):
print(file)
上面的代码使用了一个for循环遍历当前目录下的所有文件和文件夹。然后，我们使用了一个if语句来检查文件名是否以”.py”结尾。如果条件成立，我们就会打印出这个文件的名字。
注意事项
- os.listdir()返回的是一个列表，其中包含指定目录中的所有文件和子目录。请注意，不包括”.”和”..”，因为它们是特殊目录。
- 如果你使用了相对路径，请确保你的当前工作目录正确，否则你可能会得到一个不正确的列表。你可以使用os.getcwd()方法来获取当前工作目录。
- 请注意，如果你尝试在一个不存在的目录中使用os.listdir()，你将会得到一个FileNotFoundError异常。
结论
os.listdir()是Python 3中一个强大和方便的方法，它能够帮助我们快速地列出指定目录中的所有文件和子目录，并可以轻松地过滤出指定类型的文件。只需要遵循以上注意事项，就可以使用os.listdir()轻松地处理文件和目录了！