Итак, идея в том, что я пытаюсь удалить цикл for в python с помощью Numpy
return np.array([points[closest==k].mean(axis=0) for k in range(k)])
Где points - это массив точек в моем случае 2d points.Ближайшим является массив меток, отслеживающий метки для каждой точки в точках.K - это просто число
Так что моя идея состоит в том, что если бы я создал массив, который был 3-мерным, который был разделен для каждого k, то я мог бы просто взять среднее значение этого массива, не делая цикл for.
Если у вас есть идея лучшей идеи, не стесняйтесь поделиться :)