在系统中的任何位置找到具有特定扩展名的文件

Question

我需要为我在Python 3.9中定义的函数编写一个测试代码。由于函数应该接收一个具有特定扩展名及其路径的文件，然后在Python中加载它，所以我的测试应该查找该扩展名的任何可用文件，并尝试正确加载它。我如何对此进行编码？

Answer 1

也许这会有所帮助：

import os
file_path = '/' # Location of files
file_extension = 'xlsx'
for i in os.listdir(file_path):
if file_extension == i.split('.')[-1]:
print(i)  # The task you want to perform with the specific files

这里file_path是文件列表的路径，file_extension是我们正在搜索的特定扩展名。