make pass keyword arg to dataframe.drop

[cdsc_reddit.git] / clustering / clustering_base.py

diff --git a/clustering/clustering_base.py b/clustering/clustering_base.py
index 98a260e9aca97e455f1fb48201a2880ef20e332c..2f37b686660ab077e879b8f092697d1652b85a9c 100644 (file)
--- a/clustering/clustering_base.py
+++ b/clustering/clustering_base.py
@@ -68,7 +68,7 @@ class clustering_job:
     def read_distance_mat(self, similarities, use_threads=True):
         print(similarities)
         df = pd.read_feather(similarities, use_threads=use_threads)
     def read_distance_mat(self, similarities, use_threads=True):
         print(similarities)
         df = pd.read_feather(similarities, use_threads=use_threads)

-        mat = np.array(df.drop('_subreddit',1))
+        mat = np.array(df.drop('_subreddit',axis=1))

         n = mat.shape[0]
         mat[range(n),range(n)] = 1
         return (df._subreddit,1-mat)
         n = mat.shape[0]
         mat[range(n),range(n)] = 1
         return (df._subreddit,1-mat)