我有一个zip文件夹，其中包含三个类别的多维RGB图像。我需要创建一个带有两个数组的pickle文件，将图像作为第一个数组，并将每个类的对应输出作为第二个数组，第一个类将为0，第二个类将为1，第三个将为2。文件需要输入图像数组作为输入，并按照X和y等类别将对应的数组作为输出进行分类。
我实现了将所有图像放在一个数组下的第一部分，但第二部分需要方向。
import glob
filelist = glob.glob('/folder_name/*.jpg')
import numpy as np
from PIL import Image
import pickle
x = np.array([np.array(Image.open(name)) for name in filelist])
filehandle = open('filename.pkl', 'wb')
pickle.dump( x, filehandle, protocol=2 )
filehandle.close()
预期结果是一个包含所有图像的numpy数组以及相应类的咸菜文件。