change fillna to dropna

2021-06-25 20:35:54 +02:00 · 2021-06-25 20:35:54 +02:00 · 754e8dfb4f
commit 754e8dfb4f
parent 84a8fb17e2
1 changed files with 1 additions and 1 deletions
--- a/kMedoids.py
+++ b/kMedoids.py
@ -43,7 +43,7 @@ def is_finished(old_medoids, new_medoids):

 def kmedoids(num_samples, num_clusters):
    df = pd.read_csv('CC GENERAL.csv', index_col='CUST_ID')
-    df = df[:num_samples].fillna(0)
+    df = df[:num_samples].dropna()
    df_scaled = pd.DataFrame(StandardScaler().fit_transform(df))

    # initialize medoids (at random)