一尘不染

如何检查文件的扩展名?

python

我正在某个程序上工作,根据文件的扩展名,我需要做不同的事情。我可以用这个吗?

if m == *.mp3
   ...
elif m == *.flac
   ...

阅读 178

收藏
2021-01-20

共1个答案

一尘不染

假设m是一个字符串,可以使用endswith

if m.endswith('.mp3'):
...
elif m.endswith('.flac'):
...

要不区分大小写,并消除可能很大的else-if链:

m.lower().endswith(('.png', '.jpg', '.jpeg'))
2021-01-20