Python listdir函数|极客教程

Python listdir函数

1. 介绍

在Python中，listdir函数是一个非常常用的函数，它用于返回指定目录下的所有文件和目录的名称列表。本文将详细介绍这个函数的使用方法，并给出一些示例代码及其运行结果。

2. listdir函数的语法

listdir函数的语法如下所示：

os.listdir(path)

path：必需，表示要返回文件和目录列表的路径。它可以是相对路径或绝对路径，也可以是文件或目录的路径。

3. 使用示例

下面是一些使用listdir函数的示例代码及其运行结果。

示例1：返回指定目录下的所有文件和目录的名称列表

import os

# 指定目录
path = "C:/users"

# 返回文件和目录名称列表
file_list = os.listdir(path)

# 打印文件和目录名称列表
print(file_list)

运行结果：

['Public', 'Default', 'Administrator', 'Guest', 'user1', 'user2', ...]

在这个示例中，我们指定了目录为”C:/users”，然后使用listdir函数返回了该目录下的所有文件和目录的名称列表。最后，我们将列表打印出来。

示例2：返回当前目录下的所有文件的名称

import os

# 返回当前目录下的文件名称列表
file_list = os.listdir()

# 打印文件名称列表
print(file_list)

运行结果：

['test.py', 'demo.txt', 'dir1', 'dir2', ...]

在这个示例中，我们没有指定path参数，因此默认返回的是当前目录下的所有文件和目录的名称列表。

示例3：返回目录中的文件的名称列表

import os

# 指定目录
path = "C:/data"

# 返回目录中文件的名称列表
file_list = [f for f in os.listdir(path) if os.path.isfile(os.path.join(path, f))]

# 打印文件名称列表
print(file_list)

运行结果：

['data1.csv', 'data2.csv', 'data3.txt', ...]

在这个示例中，我们定义了一个列表推导式来过滤出目录中的文件，并将文件的名称添加到列表中。我们使用os.path.isfile函数来判断路径是否是文件。这样，我们就可以只返回目录中的文件的名称列表。

4. 注意事项

使用listdir函数时，注意路径的格式。在Windows系统中，路径的格式是使用反斜杠（\）分隔的，而在Linux或Mac系统中，路径的格式是使用斜杠（/）分隔的。
如果指定的目录不存在，listdir函数将会引发一个OSError异常。因此，在使用该函数前，我们应该确保指定的路径是存在的。
listdir函数返回的是文件和目录的名称列表，而不是它们的绝对路径。如果需要获取绝对路径，可以使用os.path.join函数来拼接路径。