применение kmeans к различным группам данных

Question

применение kmeans к различным группам фрейма данных

Я хочу применить kmeans путем группировки на основе 'ft_id'.

Сначала создайте функцию kmeans,

I применил это с 'groupby', но я получаю ошибку.

Помогите мне.

from sklearn.cluster import KMeans

kmeans = KMeans(n_clusters=2,random_state=0)
x = np.array(round(df['kphB'],2))
def skmean(kmeans,x): 
    X = X.reshape(-1,1)
    kmeans.fit(X)
    return(kmeans.predict(X))

df.groupby('ft_id').apply(lambda x:skmean(kmeans,x)).reset_index()

введите описание изображения здесь

введите описание изображения здесь

введите описание изображения здесь