python判断一个名字是文件名还是图片名,如果是图片,则得到图片名字
时间: 2024-03-13 22:45:23 浏览: 217
python的图片识别
你可以使用Python的os模块来判断一个名字是文件名还是图片名,如果是图片,则可以使用Python的os.path模块来得到图片名字。
以下是一个示例代码:
```python
import os
filename = "example.png"
# 判断是否为图片
if filename.endswith(".jpg") or filename.endswith(".jpeg") or filename.endswith(".png") or filename.endswith(".gif"):
# 获取图片名字
basename = os.path.basename(filename)
print("这是一张图片,名字为:", basename)
else:
print("这是一个文件名。")
```
在这个示例中,我们首先使用`endswith`方法来判断给定的文件名是否以`.jpg`,`.jpeg`,`.png`或`.gif`结尾。如果是,则说明它是一张图片,我们使用`os.path.basename`方法来获取图片名字。如果不是,则说明它是一个文件名。
阅读全文