import numpy as np

def group_array(arr): group_num = 1 group_dict = {} for i in range(arr.shape[0]): if i not in group_dict.keys(): group_dict[i] = group_num for j in range(i+1, arr.shape[0]): if abs(arr[i,0]-arr[j,0])<35 and abs(arr[i,1]-arr[j,1])<16: if j not in group_dict.keys(): group_dict[j] = group_num group_num += 1 return np.append(arr, np.array(list(group_dict.values()))[:,np.newaxis], axis=1)

测试

arr = np.random.randint(0, 100, size=(1000, 2)) result = group_array(arr) print(result)