cartodb4/cartodb-postgresql
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Use anyarray for equalintervalbins

Browse Source
This commit is contained in:
Paul Norman 2016-04-27 16:10:01 -07:00
parent b19a5fc3dc
commit 956e56cd37
1 changed files with 5 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
scripts-available/CDB_EqualIntervalBins.sql
View File

@ -1,8 +1,8 @@
-- --
-- Calculate the equal interval bins for a given column -- Calculate the equal interval bins for a given column
-- --
-- @param in_array A numeric or double precision array of numbers to determine the best -- @param in_array An array of numbers to determine the best
-- to determine the bin boundary -- bin boundary
-- --
-- @param breaks The number of bins you want to find. -- @param breaks The number of bins you want to find.
-- --
@ -10,7 +10,8 @@
-- Returns: upper edges of bins -- Returns: upper edges of bins
-- --
-- --
CREATE OR REPLACE FUNCTION CDB_EqualIntervalBins ( in_array NUMERIC[], breaks INT ) RETURNS NUMERIC[] as $$
CREATE OR REPLACE FUNCTION CDB_EqualIntervalBins ( in_array anyarray, breaks INT ) RETURNS anyarray as $$
WITH stats AS ( WITH stats AS (
SELECT min(e), (max(e)-min(e))/breaks AS del SELECT min(e), (max(e)-min(e))/breaks AS del
FROM (SELECT unnest(in_array) e) AS p) FROM (SELECT unnest(in_array) e) AS p)
@ -20,12 +21,4 @@ SELECT array_agg(bins)
FROM stats) q; FROM stats) q;
$$ LANGUAGE SQL IMMUTABLE; $$ LANGUAGE SQL IMMUTABLE;
CREATE OR REPLACE FUNCTION CDB_EqualIntervalBins ( in_array double precision[], breaks INT ) RETURNS double precision[] as $$ DROP FUNCTION IF EXISTS CDB_EqualIntervalBins( numeric[], integer);
WITH stats AS (
SELECT min(e), (max(e)-min(e))/breaks AS del
FROM (SELECT unnest(in_array) e) AS p)
SELECT array_agg(bins)
FROM (
SELECT min + generate_series(1,breaks)*del AS bins
FROM stats) q;
$$ LANGUAGE SQL IMMUTABLE;