导入文件时不知道名称

Question

在同一文件夹中，我有一个脚本（在python中），该脚本需要导入一个文档，直到现在，该文档具有.csv扩展名（每次文件名都不相同）。我该怎么办？

我尝试使用：data = pd.read_csv(r'./*.csv')，但是它不起作用。

Answer 1

如果您确定CSV文件存在，请使用iglob方法：

from glob import iglob
df = pd.read_csv(next(iglob('*.csv')))

glob模块可用于通配符匹配。 iglob特别会返回一个迭代器；您要做的只是提取它产生的第一个值，然后将其传递给read_csv。

Answer 2

如果您打算对文件做更多操作并需要它们的路径，建议您使用pathlib模块中的path.glob。

from pathlib import Path

import pandas as pd


csv_files = Path.cwd().glob('*.csv')
df = pd.read_csv(next(csv_files))

csv_files是.csv文件完整路径的生成器。这类似于iglob，但只返回文件名。