Добавить столбцы в фрейм данных K-Mean - Python

Question

Я использую этот код для анализа K-Mean на кадре данных. Я могу построить результирующий фрейм данных, но хочу добавить к нему исходные столбцы фреймов. Мой питон не блестящий, поэтому он может быть простым, но я продолжаю получать неверный синтаксис для всего, что я пытаюсь.

import pandas as pd
import matplotlib.pyplot as plt
import seaborn as sns
from sklearn.cluster import KMeans
from sklearn.metrics import silhouette_samples

df2 = df[['x','y']]
k = KMeans(n_clusters=3)
assignedClusters = k.fit(df2)

sh_df = pd.DataFrame(
    {
        'silhouette': silhouette_samples(df2, assignedClusters.labels_),
        'cluster': assignedClusters.labels_
    }
)

У sh_df есть два столбца, значения «силуэт» и «кластеры». Я хочу, чтобы этот df также включал значения x и y из моего исходного df. Возможно ли это, и может кто-нибудь сказать мне, как это сделать?

Answer 1

Просто сделай

sh_df['x'] = df2['x']
sh_df['y'] = df2['y']

при условии, что sh_df и my_df упорядочены одинаково. То есть, что n-я строка одного кадра данных соответствует n-й строке другого.