Question

如何允许strExtension的格式如.pdf和.html。如果更改为strExtension =“。*”，则该脚本无效。如何使用两种格式？ pdf和html吗？

# extension of files
strExtension = ".pdf" 
# for each file in source directory
for file in os.listdir(sourceDir):
    # if file ends with extension
if file.endswith(strExtension):

Answer 1

只需测试两种情况：

if file.endswith(strExtension1) or file.endswith(strExtension2):

或如乔恩·克莱门茨（Jon Clements）在下面评论：

if file.endswith((strExtension1,strExtension2)):

Answer 2

str.endswith()不支持通配符。但是它支持元组，所以您可以这样做：

extensions = (".html", ".pdf")
if file.endswith(extensions):
    # do stuff

Answer 3

这对我有用。感谢您的快速帮助。谢谢大家...

# extension of files
strExtension1 = ".pdf"
strExtension2 = ".html" 
# for each file in source directory
for file in os.listdir(sourceDir):
# if file ends with extension
if file.endswith(strExtension1) or file.endswith(strExtension2):

strExtension-允许两种格式

3 个答案: