使用Python打开未知的Excel文件名

Question

我正在开发一个Python脚本，它可以在不知道文件名的情况下从特定目录打开Excel文件。目录是固定和控制的，这样用户就可以到达文件所在的目录。每个目录中只有一个Excel文件。是否可以使用xlrd open命令打开Excel文件，而不指定给定目录中的文件名？在

或者，有没有一种方法可以绕过这个问题，让一个函数在给定的目录中搜索.xlsx扩展名，然后将这些文件名存储在一个数组中，这样xlrd.open_workbook()就可以使用数组中的元素打开该文件了？在

Answer 1

下面是一个简单的函数，它包装open_workbook()，使用^{}将folder中的任意.xlsx文件传递给它：

import os.path
from glob import glob
from xlrd import open_workbook

def open_arbitrary_workbook(folder, *args, **kwargs):
    try:
        path = glob(os.path.join(folder, "*.xlsx"))[0]
    except IndexError:
        raise IOError("No .xlsx files found in %r" % folder)
    return open_workbook(path, *args, **kwargs)