如何按两个键对多维数组进行排序

def custom_compare(item):
    # 自定义比较函数，按照键1和键2的值进行比较
    key1 = item[0]  # 假设键1在数组的第一个位置
    key2 = item[1]  # 假设键2在数组的第二个位置
    if key1 < key2:
        return -1
    elif key1 > key2:
        return 1
    else:
        return 0

def multi_dimensional_array_sort(arr):
    # 使用快速排序算法对多维数组进行排序
    if len(arr) <= 1:
        return arr
    pivot = arr[len(arr) // 2]
    left = [x for x in arr if custom_compare(x) < 0]
    middle = [x for x in arr if custom_compare(x) == 0]
    right = [x for x in arr if custom_compare(x) > 0]
    return multi_dimensional_array_sort(left) + middle + multi_dimensional_array_sort(right)

# 示例数据
arr = [[1, 2], [3, 4], [2, 5], [1, 3], [3, 1]]

# 调用排序函数对多维数组进行排序
sorted_arr = multi_dimensional_array_sort(arr)

# 输出排序结果
print(sorted_arr)