move back to colnames

2016-11-15 12:02:42 +01:00 · 2016-11-15 12:02:42 +01:00 · 0d40080f6c
commit 0d40080f6c
parent 0867e69d1f
1 changed files with 1 additions and 3 deletions
--- a/src/py/crankshaft/crankshaft/clustering/kmeans.py
+++ b/src/py/crankshaft/crankshaft/clustering/kmeans.py
@ -92,9 +92,7 @@ def _extract_columns(db_resp, id_col_name):
        id_col_name (string): name of column which has the row id (not a
                              feature of the analysis)
    """
-    keys = [k for k in db_resp[0].keys()]
-
-    return np.array([db_resp[0][c] for c in keys
+    return np.array([db_resp[0][c] for c in db_resp.colnames()
                     if c != id_col_name],
                    dtype=float).T