用于输出自定义分类法而不是输出单词“array”的简短代码

# 定义自定义分类法
def custom_classification():
    categories = {
        1: "分类1",
        2: "分类2",
        3: "分类3"
    }
    return categories

# 输出自定义分类法
def output_custom_classification(categories, input_text):
    if input_text == "array":
        # 返回分类1
        return categories[1]
    else:
        # 返回分类2
        return categories[2]

# 测试代码
categories = custom_classification()
input_text = "array"
output = output_custom_classification(categories, input_text)
print(output)