在2Dnumpy数组中创建作为group-by样式查找的字典的最快方法？

import numpy as np

def create_groupby_dict(arr):
    unique_vals = np.unique(arr[:, 0])  # 找到第一列的唯一值
    split_arrs = np.split(arr[:, 1:], np.where(np.diff(arr[:, 0]) != 0)[0]+1)  # 根据第一列的唯一值进行拆分
    groupby_dict = {val: split_arr for val, split_arr in zip(unique_vals, split_arrs)}  # 创建字典
    return groupby_dict

# 示例用法
arr = np.array([[1, 2, 3],
                [1, 4, 5],
                [2, 6, 7],
                [2, 8, 9]])

result = create_groupby_dict(arr)
print(result)

{1: array([[2, 3],
           [4, 5]]), 
 2: array([[6, 7],
           [8, 9]])}