将io.imread应用于数据框列

时间:2018-11-20 22:00:29

标签: python pandas numpy

我有一个数据框df,其中的“海报”列包含指向jpeg图像的链接。如果我申请

img_data = io.imread(url)

theres没问题,因为我使用的是特定网址。

我想做这样的事情:

x = np.asarray(io.imread(df['Poster']).tolist())

并获得一个4维的numpy数组(图像数,h,w,通道数)

1 个答案:

答案 0 :(得分:0)

尝试一下:

posters = []
for poster in df['Poster']:
    posters.append(np.asarray(io.imread(poster).tolist()))

更奇特的方式:

posters = [np.asarray(io.imread(poster).tolist()) for poster in df['Poster']]