代码思路怎么想的，不会

2.该明星脸部图像12张(png格式），每张都不

相同，文件名为1.png、2.png…...，图像尺寸

为100×100，图像尽量保持正面

3.图像文件分别保存在train_set和test_set两个

目录中，每个目录都有一个“张三”文件夹

（“张三”是明星名字，不要双引号）

5.“train_set/张三”目录下有10张图像

  1. “test _set/张三”目录下有2张图像

大佬，想问一下，这个要怎么写代码？

import os

data = []
a_list = ['train_set','test_set']

for i in range(0,len(a_list)):
    
    for file in os.listdir(os.getcwd()+'\\'+a_list[i]):
        excel_path = os.path.join(os.getcwd()+'\\'+a_list[i],file)
        print(excel_path)
        
        for filename in os.listdir(excel_path):
            lena = os.path.join(excel_path+'\\'+filename)
            print(lena)
            
            data.append(lena)

img