使用 pathlib 查找单个 csv 文件的最简单方法

如何解决使用 pathlib 查找单个 csv 文件的最简单方法

我想使用 csv 读取子文件夹 foo 中唯一的 pahtlib 文件，而无需明确指定名称。这样，如果文件名改变，代码仍然有效，但前提条件是 csv 中只有一个 foo 文件仍然有效。

现在我愿意

from pathlib import Path

foo_dir = Path.cwd() / 'foo'
for file in foo_dir.glob('*'):
    if file.is_file() & (file.suffix = '.csv'):
        csv_file = file
        break 

这是...呃，它有效，但有点罗嗦。在不牺牲可读性的情况下，有什么我可以做的更简单的事情吗？

PS 如何修改上面的代码，以便在 csv 中没有 foo 文件的情况下，它会引发错误？

解决方法

pathlib.Path.glob 是一个生成 pathlib.Path 对象的生成器，因此您可以在其上调用 next 以仅使用生成器中的第一个路径对象。您可以为 next 提供一个可选参数，以防止在不存在 CSV 文件时抛出 StopIteration 异常（在这种情况下，next 返回 None）。>

from pathlib import Path

if (path := next(Path("dir/to/file").glob("*.csv"),None)) is None:
    print("No .csv file present!")
else:
    print(path)

---

或

from pathlib import Path

try:
    path = next(Path("dir/to/file").glob("*.csv"))
except StopIteration:
    raise RuntimeError("No .csv file present!")

,

我会使用这样的东西

from glob import glob

try:
    csv_file = glob('foo/*.csv')[0]
except IndexError:
    raise RuntimeError('No csv file found')