python判断一个名字是文件名还是图片名
时间: 2023-12-29 18:04:28 浏览: 23
Python中可以通过判断文件名的后缀来判断它是不是图片名。通常情况下,常见的图片后缀有`.jpg`、`.jpeg`、`.png`、`.bmp`等。
下面是一个简单的示例代码,可以判断一个文件名是不是图片名:
```python
import os
def is_image(filename):
name, ext = os.path.splitext(filename)
ext = ext.lower()
return ext in ('.jpg', '.jpeg', '.png', '.bmp')
```
该函数使用了os模块的`os.path.splitext`函数来获取文件名和扩展名,然后判断扩展名是否是图片格式。`ext.lower()`是为了防止扩展名大小写不一致的问题。