我已经设置了一个包含图像路径的Pandas Dataframe,我需要向其中添加一列,其中每个单元格都应该包含一个多维数组(表示该图像)。
下面是一个例子:
import pandas as pd
import numpy as np
df = pd.DataFrame(data=[["test.png","dog.png"],[3,4]], columns=["path","B"])
# creating a new empty column
df = df.assign(image=np.nan)
image = # reading image path from row 1
df.iloc[1, df.columns.get_loc("image")] = image
但我不断得到错误:ValueError: Must have equal len keys and value when setting with an ndarray
.
我该如何解决这个问题?我已经尝试过跟踪this,但它对我不起作用。
需要说明的是,在我的真实 Dataframe 中,第n行的image字段取决于第n行的path的值。
预期结果:
path B image
0 "test.png" 2 NaN
1 "dog.png" 4 [[1,2,...], [255,255,...], ...]
1条答案
按热度按时间xqkwcwgp1#
使用
PIL.Image
模块获取可转换为数组的图像对象:输出示例: