如何根据两个元素对子数组的元素进行分组

def group_subarrays_by_elements(arr, element1, element2):
    grouped_subarrays = {}
    
    for subarray in arr:
        key = (subarray[element1], subarray[element2])
        
        if key in grouped_subarrays:
            grouped_subarrays[key].append(subarray)
        else:
            grouped_subarrays[key] = [subarray]
    
    return grouped_subarrays

# 原始数组
array = [
    [1, 2, 3],
    [4, 5, 6],
    [1, 2, 7],
    [4, 5, 8]
]

# 指定的两个元素索引
element1_index = 0
element2_index = 2

# 根据指定的两个元素对子数组进行分组
result = group_subarrays_by_elements(array, element1_index, element2_index)

# 输出分组结果
for key, value in result.items():
    print(f"键: {key}, 值: {value}")

键: (1, 3), 值: [[1, 2, 3]]
键: (4, 6), 值: [[4, 5, 6]]
键: (1, 7), 值: [[1, 2, 7]]
键: (4, 8), 值: [[4, 5, 8]]