将图像添加到熊猫数据帧行

Question

我想从文件夹中加载图像，将它们转换为numpy数组，然后将它们存储为数据帧行。我应该如何继续？以下是我想到的：from keras.preprocessing.image import load_imgfrom keras.preprocessing.image import img_to_arrayfrom keras.preprocessing.image import array_to_imgBASE_DIR = '/kaggle/input/coughspectrograms/content/'os.chdir('/kaggle/input/coughspectrograms/content/')files_in_train = sorted(os.listdir(BASE_DIR))files_in_train.pop(0)img_df = pd.DataFrame()for img in files_in_train:     loaded_img = load_img(img)    img_array = img_to_array(loaded_img)    img_df = pd.DataFrame({"image": [img_array]})

Answer 1

如果要具有形状 [n_images， 1] 的最终数据帧，只需将代码中的最后一行替换为：

img_df = img_df.append({"image": [img_array]}, ignore_index=True)

这将以迭代方式向您创建的空数据帧添加一行，您现在所做的只是在每次迭代时使用单个图像创建一个新数据框