cartodb4/crankshaft
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Merge branch 'adds-nonspatial-kmeans' of https://github.com/CartoDB/crankshaft into adds-nonspatial-kmeans

Browse Source
This commit is contained in:
Andy Eschbacher 2016-10-18 21:14:09 -06:00
commit 3c6d73b7e2
1 changed files with 5 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
src/py/crankshaft/crankshaft/clustering/kmeans.py
View File

@ -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: