python 获取文件夹下的文件名操作(两种)

一、相对路径

# 文件存储的想对路径路径
path = './data/0'
# os.walk()是一个生成器，返回三个值：根目录，根目录下的目录和文件列表
folder = os.walk(path)
"""  folder 是 <generator object walk at 0x000001DE290C6748> """"
# 将返回值转化成列表，
files = list(folder)
""" files 是 [('./data/0', [], ['01.txt', '02.txt', '03.txt', '04.txt', '05.txt'])] """
file_list = files[0][2]
""" file_list  ['01.txt', '02.txt', '03.txt', '04.txt', '05.txt']  """

二、绝对路径

"""获取该.ipynb文件的绝对路径"""
rootPath = os.getcwd()
print("根目录路径 : ",rootFilePath)

output:
根目录路径 :  C:\Users\WJL\NLP_data_list

"""获取和该文件的同一文件夹下的data文件的绝对路径"""
data_path = os.path.join(rootFilePath, 'data')
print("data_path : ",data_path)

output:
data_path :  C:\Users\WJL\NLP_data_list\data

"""获取data文件下所有文件的名称列表"""
data_listdir_name = os.listdir(data_path)
print("data_listdir_name : ",data_listdir_name)

output:
data_listdir_name :  ['0', '1', '2']

"""获取data文件下所有文件的绝对路径列表列表"""
data_listdir_path = [os.path.join(data_path,name) for name in data_listdir_name]
data_listdir_path

output:
['C:\\Users\\WJL\\NLP_data_list\\data\\0',
 'C:\\Users\\WJL\\NLP_data_list\\data\\1',
 'C:\\Users\\WJL\\NLP_data_list\\data\\2']

"""获取0文件夹的所有文件名称列表"""
data_0_listdir_name = os.listdir(data_listdir_path[0])
data_0_listdir_name

output:
['01.txt', '02.txt', '03.txt', '04.txt', '05.txt']

总结：非特殊情况，习惯用第一种，也推荐用第一种。