读取没有全名的 CSV 文件

Question

Python 在这里非常新。 我正在尝试从文件夹中读取文件，但不知道其全名。 例如AAA_05212021.csv是位于C:\test\中的文件名 AAA 是文件名的已知部分。 rest 每天都在变化。

我试过了：

import pandas as pd
data = pd.read_csv(r'C:\test\AAA*.csv')

..但它似乎不起作用。 任何帮助将非常感激。 谢谢！

Answer 1

您可以为此使用glob 。 Glob 将返回与该模式匹配的文件列表。 由于目录中只有一个文件与模式匹配，因此您可以使用索引0获取它：

from glob import glob
import pandas as pd

file = glob('C:\test\AAA*.csv')[0]
data = pd.read_csv(file)

Answer 2

此代码应识别所有可能的文件：

import os

valid_files = []
for file in os.listdir('C:\test'):
    # Check that "AAA" is in the filename
    if "AAA" in file: 
        # Check that "AAA" is at the start of the filename
        if file.partition("AAA")[0] == "":
            valid_files.append(file)

当然，如果文件夹中有多个遵循该结构的文件，您仍然需要以某种方式选择一个，但如果您不挑剔，那可能就像files[0]一样简单。