diff --git a/src/py/crankshaft/crankshaft/clustering/kmeans.py b/src/py/crankshaft/crankshaft/clustering/kmeans.py index 21d76ea..0df9ce0 100644 --- a/src/py/crankshaft/crankshaft/clustering/kmeans.py +++ b/src/py/crankshaft/crankshaft/clustering/kmeans.py @@ -8,13 +8,11 @@ def kmeans(query, no_clusters, no_init=20): find centers based on clusteres of latitude/longitude pairs query: SQL query that has a WGS84 geometry (the_geom) """ - full_query = ''' - SELECT array_agg(cartodb_id ORDER BY cartodb_id) as ids, - array_agg(ST_X(the_geom) ORDER BY cartodb_id) xs, - array_agg(ST_Y(the_geom) ORDER BY cartodb_id) ys - FROM ({query}) As a - WHERE the_geom IS NOT NULL - '''.format(query=query) + full_query = ("SELECT array_agg(cartodb_id ORDER BY cartodb_id) as ids," + "array_agg(ST_X(the_geom) ORDER BY cartodb_id) xs," + "array_agg(ST_Y(the_geom) ORDER BY cartodb_id) ys " + "FROM ({query}) As a " + "WHERE the_geom IS NOT NULL").format(query=query) try: data = plpy.execute(full_query) except plpy.SPIError, err: