获取一个文件夹中所有图像数据的名称并保存到CSV文件中

import os
import pandas as pd

# 定义图像文件夹的路径
image_folder = r'E:\NiuCode\LianXi\data\Chest X-ray\pneumonia'

# 获取文件夹下所有文件的名称
image_names = os.listdir(image_folder)

# 创建一个DataFrame对象，将图像名称作为一列
df = pd.DataFrame(image_names, columns=['图像名称'])

# 定义保存CSV文件的路径
csv_file_path = r'E:\NiuCode\LianXi\data\Chest X-ray\pneumonia_image_names.csv'

# 将DataFrame保存为CSV文件
df.to_csv(csv_file_path, index=False)

print(f"图像名称已成功保存到 {csv_file_path}")