时间戳

首页 > 解决方案 > 加载具有相似文件名的文件的 Pythonic 方式

python - 加载具有相似文件名的文件的 Pythonic 方式

问题描述

我pathlib.Path()用来检查文件是否存在，并使用rasterio将其作为图像打开。

filename = pathlib.Path("./my_file-name.tif")

但是，我从中加载的目录中的某些文件的文件名略有不同，例如 my_filename.tif（不带连字符）或my_file_name.tif（带下划线而不是连字符）。

所有文件名都具有相同的基本结构。除了检查它们是否都存在之外，有没有更好的方法来调用pathlib.Path()所有可能的变化？例如，filename.tif

filename = pathlib.Path("./my_file-name.tif")
if not file.is_file():
    file = Path("./my_file_name.tif")
if not file.is_file():
    file = Path("./my_filename.tif")

标签： pythonpython-3.xpathlib

解决方案

您可以使用正则表达式（或表达式）来查看目录中的任何文件是否匹配。但是您也可以只检查文件名是否包含所有必要的组件。这是您可以调整的示例：

dir = pathlib.Path("./")
must_contain = ['my', 'file', 'name', '.tif']
true_filepath = None
for filepath in os.listdir(dir):
    if all([item in filepath for item in must_contain]):
        true_filepath = filepath
        break

这将选择包含所有必要组件的第一个项目。如果您对组件必须采用的形式有特定要求，那么我建议您使用正则表达式路线，您可以在其中使用符合您需要的正则表达式。这是python 3 中正则表达式包re的文档。

希望这会有所帮助，快乐编码！

推荐阅读