python - 将条目从 .csv 文件转换回图形的方法
问题描述
我正在尝试使用 Image 将图形信息保存到 .csv 文件,但我坚持将其转换回图形。它不断给我错误“AttributeError:'str'对象没有属性' array_interface '”。
我想这意味着我从 .csv 文件中提取的条目是一个字符串,需要转换为数组?
我将图形转换为 np 数组的代码如下所示:
from PIL import Image
img = np.array(Image.open(fig_file))
file_name = 'data.csv'
row_contents = [labels, img]
from csv import writer
def append_list_as_row(file_name, list_of_elem):
# Open file in append mode
with open(file_name, 'a+', newline='') as write_obj:
# Create a writer object from csv module
csv_writer = writer(write_obj)
# Add contents of list as last row in the csv file
csv_writer.writerow(list_of_elem)
append_list_as_row(file_name, row_contents)
有问题的部分(将其转换回图形)如下所示:
import pandas as pd
df1 = pd.read_csv(file_name)
fig_array = df1.loc[1, "img"]
img = Image.fromarray(fig_array, 'RGB')
img.save('test.png')
图像行导致错误。也许我不应该使用熊猫来定位条目?有修改的想法吗?我试过 .to_numpy(),它不起作用。
太感谢了!
解决方案
首先,如果可能的话,不要这样做。这成本太高了。只需制作一个表格(数据框),记录与每个文件关联的标签,以便以后查询。例如
| file_id | file_path | label |
|---------|-----------|-----------|
| 1 | a.jpg | fine-arts |
| 2 | b.png | manga |
| 3 | c.jpb | whatever |
如果您真的必须将图像编码为字符串,那么base64 编码是一种常见的方法。例如,jupyter notebook
使用base64
格式将图像嵌入到 html 文件中,以便用户可以轻松地共享生成的图像。
其次,由于电子表格软件的列宽限制,仍然不建议将(标签,数据)对保存为 csv 文件。如果不能利用.csv
格式,为什么要使用它?因此,在这种情况下,最好还是做一个上面提到的查找表,以避免不必要的代价高昂的转换。
如果你还在做,好的,这里是示例代码。SMALL图像取自debian 主页。可以验证数据是否正确恢复。
代码:
import numpy as np
from PIL import Image
import base64
import csv
# https://www.debian.org/Pics/openlogo-50.png
img_path = "/mnt/ramdisk/debian-openlogo-50.png"
img = np.array(Image.open(img_path))
img_encoded = base64.b64encode(img).decode("ascii")
label = "fine-arts"
# Simulate multiple records
data = [
[label, img_encoded],
[label, img_encoded],
[label, img_encoded]
]
# save
with open("/mnt/ramdisk/sav.csv", "w+") as f:
w = csv.writer(f)
w.writerows(data)
# load
data_loaded = []
with open("/mnt/ramdisk/sav.csv") as f:
r = csv.reader(f)
for row in r:
data_loaded.append(row)
# check data are unchanged after S/L
for i in range(3):
for j in range(2):
assert data[i][j] == data_loaded[i][j]
# decode the image (still need shape info)
r = base64.b64decode(data_loaded[0][1].encode("ascii"))
img_decoded = np.frombuffer(r, dtype=np.uint8).reshape((61, 50, 4))
# check image is restored correctly
import matplotlib.pyplot as plt
plt.imshow(img_decoded)
plt.show()
但是,如果使用蒙娜丽莎等较大的图像,则 csv 阅读器会抱怨:
_csv.Error: field larger than field limit (131072)
而且您仍然需要图像形状来恢复尺寸。因此,实际上需要第三列存储图像形状。
推荐阅读
- grails - “每个入”/“无入”的 Gorm 标准
- c# - 如何在同一wifi中连接其他PC上的网络服务器?
- python - 如何以编程方式更新python文件中的文档字符串
- python-3.x - Python3 中的 ASCII 问题
- regex - 正则表达式:想要消除破折号开始的情况
- laravel - Laravel - 如何将 laravel 项目从 MacBook 移动到 PC
- sql-server - 如何检查 SQL 记录是否按特定顺序
- java - 从具有多个 SSN 的文件中仅使用部分掩码屏蔽所有 SSN
- javascript - 在“updated_cart_totals”jQuery 事件中获取 WooCommerce 刷新的购物车总数
- python - 如何从我的谷歌地图自动提取数据(KMZ?JSON?)