python判断一个名字是文件名还是图片名,如果是图片,则得到图片路径
时间: 2024-03-10 08:50:33 浏览: 222
Python中查看文件名和文件路径
你可以通过判断文件名的后缀名来判断文件类型。如果文件名以".jpg"、".png"、".jpeg"、".bmp"、".gif"等为结尾,则可以认为它是图片文件。你可以使用Python的os库中的listdir()函数,遍历指定目录下的所有文件,然后根据文件名的后缀名判断文件类型,最后得到图片的路径。下面是一个简单的示例代码:
```python
import os
# 指定目录路径
dir_path = "your_directory_path"
# 遍历目录下所有文件
for file_name in os.listdir(dir_path):
# 判断是否为图片文件
if file_name.lower().endswith(('.jpg', '.jpeg', '.png', '.bmp', '.gif')):
# 得到图片路径
image_path = os.path.join(dir_path, file_name)
print(image_path)
```
在代码中,首先指定了要遍历的目录路径,然后使用os.listdir()函数获取目录下的所有文件名。接着使用文件名的lower()方法将文件名转换为小写字母,然后判断文件名是否以图片后缀名为结尾,如果是,则使用os.path.join()函数将目录路径和文件名合并为完整的图片路径。最后打印出图片路径。
阅读全文