Python：需要使用文件名和两个扩展名中的任意一个

Question

path = "/test/"
file = "afile"
path = (path + '/' + file + ('.png' or '.jpg'))

测试是否有afile.jpg或afile.png并使用它拥有的那个。你知道吗

我知道这是不正确的，但它总结了我想做的事。你知道吗

我有一个文件名（file），想把它添加到path，因为我不确定这个文件是JPG还是PNG，所以我想两者都试试。你知道吗

显然，上述做法行不通。但我需要一张简单的支票。我的解决方案需要列出匹配的文件目录。必须有一个简单的解决方案？你知道吗

Answer 1

测试文件是否存在：

filename = os.path.join(path, file)
for extension in ('.png', '.jpg'):
    if os.path.isfile(filename + extension):
        filename += extension
        break
else:
    print 'File not found'