在 BigQuery 中从 Oracle SQL 创建 width_bucket 函数

Question

我希望通过 BigQuery 中的新函数复制 Oracle 中可用的 width_bucket 函数。该函数根据您在最小值和最大值之间指定的数字创建等宽桶。例如，width_bucket(user_count, 0, 35, 10) 将创建 10 个相等的桶，例如 0 - 3.5、3.5 - 7 等，并告诉您 user_count 属于哪个桶。任何帮助将不胜感激！

Oracle 文档 - https://docs.oracle.com/cd/B19306_01/server.102/b14200/functions214.htm

这是我所拥有的，我相信这是可行的，但我想得到它，所以如果可能的话，我不必引用表格来生成行号。

CREATE OR REPLACE FUNCTION functions.widthBucket(
  value NUMERIC,
  minValue NUMERIC,
  maxValue NUMERIC,
  buckets INT64)
AS ((
  SELECT resultBucket 
  FROM (
      SELECT CASE 
               WHEN value >= (minValue * bucketNumber) + ((maxValue/buckets) * (bucketNumber - 1))
                AND value < (maxValue/buckets) * bucketNumber 
               THEN bucketNumber
               WHEN value = maxValue and bucketNumber = buckets 
               THEN bucketNumber
             ELSE -1 
             END as resultBucket
      FROM (
            SELECT ROW_NUMBER() OVER (PARTITION BY '') as bucketNumber
            FROM project.dateTable
           ) x
      WHERE bucketNumber <= buckets) x
  WHERE resultBucket != -1
  ));

Answer 1

以下是 BigQuery 标准 SQL

试试下面 - 我认为它完全符合您的要求

CREATE TEMP FUNCTION widthBucket(
  value NUMERIC, 
  minValue NUMERIC, 
  maxValue NUMERIC, 
  buckets NUMERIC
) AS (
  RANGE_BUCKET(value, GENERATE_ARRAY(minValue, maxValue, (maxValue - minValue)/buckets))
);

使用就像你的问题一样简单 例如widthBucket(user_count, 0, 35, 10)

当值等于 maxValue 时解决边缘情况 - 使用上面的以下变化

CREATE TEMP FUNCTION widthBucket(
  value NUMERIC, 
  minValue NUMERIC, 
  maxValue NUMERIC, 
  buckets NUMERIC
) AS ((
  SELECT IF(bucket > buckets, buckets, bucket)
  FROM (
    SELECT RANGE_BUCKET(value, GENERATE_ARRAY(minValue, maxValue, (maxValue - minValue)/buckets)) bucket
  )
));