如何在循环中按类别打印预测

# 定义类别与打印函数的映射关系
category_print_mapping = {
    '类别A': print_category_a,
    '类别B': print_category_b,
    '类别C': print_category_c,
    # 添加更多类别和对应的打印函数
}

# 循环遍历预测结果
for prediction in predictions:
    category = get_category(prediction)  # 获取预测结果的类别

    # 根据类别选择打印函数
    if category in category_print_mapping:
        category_print_mapping[category](prediction)
    else:
        print("未知类别")

# 打印函数示例
def print_category_a(prediction):
    print("类别A的预测结果：", prediction)

def print_category_b(prediction):
    print("类别B的预测结果：", prediction)

def print_category_c(prediction):
    print("类别C的预测结果：", prediction)