根据键值将二维数组拆分为n个

def split_array_by_key(array, n):
    # Step 1
    key_dict = {}
    for row in array:
        key = row[0]
        if key in key_dict:
            key_dict[key].append(row)
        else:
            key_dict[key] = [row]
    
    # Step 2
    key_counts = {}
    for key in key_dict:
        key_counts[key] = len(key_dict[key])
    key_list = list(key_counts.keys())
    
    # Step 3
    if n > len(key_list):
        n = len(key_list)
    keys_per_array = len(key_list) // n
    
    # Step 4
    key_lists = [key_list[i:i+keys_per_array] for i in range(0, len(key_list), keys_per_array)]
    
    # Step 5
    result_arrays = []
    for keys in key_lists:
        sub_array = []
        for key in keys:
            sub_array.extend(key_dict[key])
        result_arrays.append(sub_array)
    
    return result_arrays