两个文件夹中一个放txt文件,另一个放图片,用python根据txt文件名打开同名的图片
时间: 2024-04-30 10:21:17 浏览: 119
可以使用Python中的os和PIL库来实现这个功能。
首先,我们需要遍历存放图片的文件夹,获取所有图片的文件名,并将其存储到一个字典中。然后,我们遍历存放txt文件的文件夹,获取所有txt文件的文件名,并根据其文件名在字典中查找对应的图片文件。
下面是一个示例代码:
```python
import os
from PIL import Image
# 存放图片的文件夹路径
img_folder = '/path/to/image/folder'
# 存放txt文件的文件夹路径
txt_folder = '/path/to/txt/folder'
# 遍历图片文件夹,获取所有图片的文件名,并将其存储到字典中
img_dict = {}
for filename in os.listdir(img_folder):
if filename.endswith('.jpg') or filename.endswith('.png'):
img_dict[filename] = os.path.join(img_folder, filename)
# 遍历txt文件夹,根据文件名查找对应的图片文件
for txt_filename in os.listdir(txt_folder):
if txt_filename.endswith('.txt'):
# 获取txt文件名(不包含扩展名)
name = os.path.splitext(txt_filename)[0]
# 在字典中查找对应的图片文件
if name in img_dict:
img_path = img_dict[name]
# 打开图片并进行相关操作
img = Image.open(img_path)
# ...
```
在上面的代码中,我们使用`os.listdir()`函数来遍历文件夹中的文件,使用`os.path.join()`函数拼接文件路径,使用`os.path.splitext()`函数获取文件名和扩展名。最后,我们使用PIL库中的`Image.open()`函数打开图片并进行相关操作。
阅读全文