pythonでk-means法を用いて、3つのグループに分類

Question

生徒200人の成績データのデータを使ってk-means法を用いて、3つのグループに分類したいです。

import pandas as pd
import matplotlib.pyplot as plt
from sklearn.cluster import KMeans # K-means クラスタリングをおこなう

df = pd.read_csv("seiseki.csv")

print(df.head())


kmeans_model = KMeans(n_clusters=3, random_state=10).fit(df)

labels = kmeans_model.labels_

print(labels)

ValueError: Input contains NaN, infinity or a value too large for dtype('float64').

kmeans_model = KMeans(n_clusters=3, random_state=10).fit(df)のところがエラー吐きます。
何かわかる方いましたら教えてください。

Comments

[meg_]

データが悪いのだと思います。エラーメッセージ「ValueError: Input contains NaN, infinity or a value too large for dtype('float64').」についてデータの中身は確認されましたか？

Answer 1

データフレームにNaNが含まれているので，欠損値の処理を行う必要があります