如何将文件夹内的图像按文件名前缀标注类别后加载生成pandas DataFrame

首先确保你已经安装了pandas库，未安装可先执行pip install pandas。

完整可运行代码 #

import os
import pandas as pd

# 替换为你的图像文件夹实际路径
img_folder = "./your_image_folder"

# 文件名前缀与类别的映射规则，匹配预期输出，可自行调整对应数值
class_map = {
    "hp": 0,
    "toshiba": 1,
    "dell": 2
}

data = []
# 遍历文件夹下的所有文件
for filename in os.listdir(img_folder):
    # 只处理png格式图像，可根据需要调整后缀判断逻辑
    if filename.endswith(".png"):
        for prefix, cls in class_map.items():
            if filename.startswith(prefix):
                data.append({"filename": filename, "class": cls})
                break

# 生成DataFrame
df = pd.DataFrame(data)
# 按类别升序排序，和示例输出格式保持一致
df = df.sort_values("class").reset_index(drop=True)

print(df)

输出效果 #

运行后输出的DataFrame和你预期的格式一致：

filename  class
0       hp_01.png      0
1  toshiba_01.png      1
2     dell_01.png      2

内容的提问来源于stack exchange，提问作者StackUser