Python中如何获取某个文件夹中以hdf5结尾的文件的名字,给出示例代码
时间: 2024-05-14 15:16:17 浏览: 84
HDF5DatasetWriter.rar_-baijiahao_hdf5_工具类_文件
可以使用Python内置的os模块和glob模块来实现。
示例代码:
```python
import os
import glob
folder_path = '/path/to/folder' # 文件夹路径
file_extension = '*.hdf5' # 文件扩展名
# 方法一:使用os模块
file_names = []
for file in os.listdir(folder_path):
if file.endswith('.hdf5'):
file_names.append(file)
# 方法二:使用glob模块
file_names = glob.glob(os.path.join(folder_path, file_extension))
print(file_names)
```
其中,方法一使用os模块的`listdir()`函数获取文件夹中所有文件的文件名,然后判断文件名是否以`.hdf5`结尾,如果是,则加入到`file_names`列表中。
方法二使用glob模块的`glob()`函数,传入文件夹路径和文件扩展名,获取所有以`.hdf5`结尾的文件的路径,返回一个列表。可以使用`os.path.basename()`函数来获取文件名。
阅读全文