我有一个excel表格数据,一共6列特征,5000行样本,我想给每个样本添加图像标签,通过下面代码实现的。利用这个方法实现的标签好像是字符型的,在后面使用时报错“could not convert string to float”,找了一些方法但一直没解决。望大家看看是什么原因。
# 读取Excel表格数据
data = pd.read_excel("data.xlsx")
# 创建一个新的列用于存储图像标签
data["label"] = None
# 遍历每一行数据
for index, row in data.iterrows():
# 获取当前行的探测器编码数据
encoding_data = list(row.values)
# 加载对应的图像
image_path = "image/{}.png".format(row['pre_label'])
image = Image.open(image_path)
# 将图像添加到对应行的"label"列中
data.at[index, "label"] = image
# 保存带有图像标签的数据
data.to_excel("data2.xlsx", index=False)