Python 使用图像创建.csv文件

假设给我一个培训文件夹。培训文件夹包含单独的文件夹，其中包含图像（CNV、DME、DRUSEN和NORMAL都是检测类型）：

我想检查每个文件夹中的每个图像，并为每个图像分配一个

• 唯一ID和疾病名称

然后将其写入CSV

我正在努力完成的示例：

---

我想要一个CSV，该CSV具有唯一id，用于一列文件夹中包含的所有60000个图像，另一列包含疾病类型（或文件：CNV、DME、DRUSEN、NORMAL）。

您可以在

os.listdir（）的帮助下完成此操作

我还没有测试过，但你有这个想法，对吧？
是的！这就是我想要做的，我也尝试过做类似的事情，但是我要花很长时间才能取消它，因为我认为我做错了。我猜我必须先运行for循环4次（每个文件夹一次），然后才能运行open代码，对吗？
csv = 'id, disease\n'
for disease in os.listdir(root_data_path):
    for file in os.listdir(os.path.join(root_data_path, disease)):
        csv += f'{file},{disease}\n'

with open('diseases_ids.csv', 'w+') as output_csv:
    output_csv.write(csv)