做一个返回分类数组的函数？

def classify_array(arr):
    classified_array = {}

    # 遍历输入数组，将元素分类
    for num in arr:
        category = num % 2  # 假设以奇偶数作为分类标准
        if category not in classified_array:
            classified_array[category] = []
        classified_array[category].append(num)

    return list(classified_array.values())

# 示例用法
arr = [1, 2, 3, 4, 5, 6]
result = classify_array(arr)
print(result)  # 输出: [[1, 3, 5], [2, 4, 6]]