以下是一个示例代码,用于遍历目录中的所有文件夹,找出其中的h5文件:
import os
def find_h5_files(directory):
h5_files = []
for root, dirs, files in os.walk(directory):
for file in files:
if file.endswith(".h5"):
h5_files.append(os.path.join(root, file))
return h5_files
directory = "path/to/directory" # 替换为要遍历的目录路径
h5_files = find_h5_files(directory)
for file in h5_files:
print(file)
在这个示例中,我们使用了os.walk()
函数来遍历目录中的所有文件夹和文件。对于每个文件,我们检查其文件名是否以".h5"结尾,如果是,则将其路径添加到h5_files
列表中。
最后,我们遍历h5_files
列表,并打印每个h5文件的路径。你可以根据自己的需求修改代码来处理这些文件,比如进行进一步的操作或者读取文件内容等。