我可以通过数据框架将图像分类到目录中吗

Question

我有一个pandas数据帧，其中第1列是我的文件路径，第2列是我的标签。是否可以根据标签的值将图像排序到新目录中。我使用3个标签，因此值为0、1、2

dataframe = pd.read_csv('\\extension.csv', header = True)
if dataframe.iat[0,1] == 0:
    if not os.path.exists(dest_fold_0):
        os.mkdir(dest_fold_0)
    img.save(dest_fold_0 + dataframe.iat[1, 0], 'jpeg')

数据帧示例：

File_Extension,Labels
['\\test_0000_50x50.jpg'],0
['\\test_0001_50x50.jpg'],1
['\\test_0002_50x50.jpg'],0
['\\test_0003_50x50.jpg'],1
['\\test_0004_50x50.jpg'],2
['\\test_0005_50x50.jpg'],0
['\\test_0006_50x50.jpg'],1