python调用文件夹中的所有格式图片,并将图片转换成二维数组的形式,将名称大小保存到表格中输出
时间: 2024-02-22 09:55:52 浏览: 124
好的,可以使用Python中的os和Pillow库来实现。以下是一个简单的代码示例:
```python
import os
from PIL import Image
import numpy as np
import pandas as pd
# 定义图片目录路径和表格文件名
image_dir = 'images/'
csv_file = 'image_info.csv'
# 支持的图片格式
image_formats = ['.jpg', '.jpeg', '.png', '.bmp', '.gif']
# 遍历图片目录
images = []
for file_name in os.listdir(image_dir):
if os.path.splitext(file_name)[1].lower() in image_formats: # 只处理指定格式的图片
file_path = os.path.join(image_dir, file_name)
with Image.open(file_path) as img:
width, height = img.size # 获取图片大小
img_array = np.array(img) # 将图片转换成二维数组
images.append({'name': file_name, 'width': width, 'height': height, 'array': img_array}) # 将图片名称、大小和数组保存到列表中
# 将图片信息保存到表格中
df = pd.DataFrame(images)
df.to_csv(csv_file, index=False)
# 输出表格内容
print(df)
```
这段代码会遍历指定目录中的所有支持的图片格式,并将图片名称、宽度、高度及对应的二维数组保存到一个列表中。然后使用Pandas库将这些信息保存到一个CSV文件中,并在控制台输出表格内容。你可以根据实际需求修改代码来适应你的项目。
阅读全文