Python中获取目录下所有文件和文件夹

在Python中，我们经常需要获取一个目录下的所有文件和文件夹，这时就需要使用到os模块和os.path模块了。

使用os.listdir方法

os.listdir方法用于获取指定目录下的所有文件和文件夹，并返回一个列表，我们可以通过循环遍历来获取每一个文件和文件夹的名称。

import os

def get_files(path):
    files = os.listdir(path)
    for file in files:
        print(file)

上面的代码会打印出指定目录下所有的文件和文件夹的名称。

使用os.walk方法

os.walk方法可以用于遍历指定目录下的所有文件和文件夹，包括子目录中的文件和文件夹。

import os

def get_files(path):
    for dirpath, dirnames, filenames in os.walk(path):
        for dirname in dirnames:
            print(os.path.join(dirpath, dirname))
        for filename in filenames:
            print(os.path.join(dirpath, filename))

上面的代码会打印出指定目录下所有的文件和文件夹的名称，包括子目录中的文件和文件夹。

使用os.path模块

os.path模块提供了一些常用的路径操作函数，如os.path.isdir、os.path.isfile等。可以通过这些函数来判断一个路径是文件还是目录。

import os

def get_files(path):
    for file in os.listdir(path):
        file_path = os.path.join(path, file)
        if os.path.isfile(file_path):
            print(file_path + " is a file")
        else:
            print(file_path + " is a directory")

上面的代码会判断指定目录下每一个文件和文件夹是文件还是目录，并打印出相应的信息。

以上就是Python中获取目录下所有文件和文件夹的方法，在实际开发中可以根据需求选择适合自己的方法来实现文件和文件夹的操作。