elasticsearch - 按百分位数过滤

时间:2018-05-04 03:57:45

标签: elasticsearch percentile

假设我想要在10到20个百分位数内的某个字段过滤文档。我想知道是否可以通过一些简单的查询,例如{"fieldName":{"percentile": [0.1, 0.2]}}

说我有这些文件:

[{"a":1,"b":101},{"a":2,"b":102},{"a":3,"b":103}, ..., {"a":100,"b":200}]

我需要按a(升序)过滤前10位,从1到10为a。然后我需要按b对这些结果进行排序按降序排列,然后取分页结果(如第2页,每页10个项目)。

一个解决方案是:

  1. 获取文件的总数。

  2. a对文档进行排序,采取相应的_id限制0.1 * total_count

  3. 编写最终查询,例如id in (...) order by b

  4. 但缺点也很明显:

      如果我们谈论亚秒延迟

    1. 似乎没有效果

    2. 如果我们在第一个查询中返回了太多_id,则第二个查询可能不起作用(默认情况下ES只允许1000个。我可以更改配置,但总是有限制)。

1 个答案:

答案 0 :(得分:1)

我怀疑如果事先不知道a的确切值,有一种方法可以在一个查询中执行此操作,尽管我认为一种非常有效的方法是可行的。

我建议先将percentiles aggregation作为第一个查询,将range query作为第二个查询。

在我的示例索引中,我只有14个文档,因此,出于解释原因,我将尝试查找字段a的30%到60%的文档,并按字段b对它们进行排序。逆序(所以要确保排序有效)。

以下是我插入的文档:

{"a":1,"b":101}
{"a":5,"b":105}
{"a":10,"b":110}
{"a":2,"b":102}
{"a":6,"b":106}
{"a":7,"b":107}
{"a":9,"b":109}
{"a":4,"b":104}
{"a":8,"b":108}
{"a":12,"b":256}
{"a":13,"b":230}
{"a":14,"b":215}
{"a":3,"b":103}
{"a":11,"b":205}

让我们找出30%到60%百分位数的字段a的界限:

POST my_percent/doc/_search
{
    "size": 0,
    "aggs" : {
        "percentiles" : {
            "percentiles" : {
                "field" : "a",
                "percents": [ 30, 60, 90 ]
            }
        }
    }
}

使用我的样本索引,它看起来像这样:

{
...
  "hits": {
    "total": 14,
    "max_score": 0,
    "hits": []
  },
  "aggregations": {
    "percentiles": {
      "values": {
        "30.0": 4.9,
        "60.0": 8.8,
        "90.0": 12.700000000000001
      }
    }
  }
}

现在我们可以使用边界来执行range查询:

POST my_percent/doc/_search
{
    "query": {
      "range": {
            "a" : {
                "gte" : 4.9,
                "lte" : 8.8
            }
        }
    },
    "sort": {
      "b": "desc"
    }
}

结果是:

{
  "took": 5,
  "timed_out": false,
  "_shards": {
    "total": 5,
    "successful": 5,
    "skipped": 0,
    "failed": 0
  },
  "hits": {
    "total": 4,
    "max_score": null,
    "hits": [
      {
        "_index": "my_percent",
        "_type": "doc",
        "_id": "vkFvYGMB_zM1P5OLcYkS",
        "_score": null,
        "_source": {
          "a": 8,
          "b": 108
        },
        "sort": [
          108
        ]
      },
      {
        "_index": "my_percent",
        "_type": "doc",
        "_id": "vUFvYGMB_zM1P5OLWYkM",
        "_score": null,
        "_source": {
          "a": 7,
          "b": 107
        },
        "sort": [
          107
        ]
      },
      {
        "_index": "my_percent",
        "_type": "doc",
        "_id": "vEFvYGMB_zM1P5OLRok1",
        "_score": null,
        "_source": {
          "a": 6,
          "b": 106
        },
        "sort": [
          106
        ]
      },
      {
        "_index": "my_percent",
        "_type": "doc",
        "_id": "u0FvYGMB_zM1P5OLJImy",
        "_score": null,
        "_source": {
          "a": 5,
          "b": 105
        },
        "sort": [
          105
        ]
      }
    ]
  }
}

请注意,percentiles汇总的结果是近似值。

通常,这看起来像是pandasSpark工作更好地解决的任务。

希望有所帮助!