Линейная интерполяция в numpy квантиль () - PullRequest
0 голосов
/ 29 февраля 2020

Рассмотрим следующий код:

>>> import numpy as np
>>> l = [21,22,24,24,26,97]
>>> np.quantile(l, 0.25)
22.5

Документация гласит:

линейный: i + (j - i) * дробь, где дробь дробная часть индекса окружена i и j.

Может ли кто-нибудь объяснить, что такое i, j и fraction в этом примере и как мы получаем 22.5?

1 Ответ

0 голосов
/ 29 февраля 2020

Работает примерно так:

import numpy as np


def my_quantile(array, q):
    n = len(array)
    index = (n - 1) * q
    if int(index) == index:  # has no fractional part
        return array[index]
    fraction = index - int(index)
    left = int(index)
    right = left + 1
    i, j = array[left], array[right]
    return i + (j - i) * fraction


if __name__ == '__main__':
    arr = [21, 22, 24, 24, 26, 97]
    assert my_quantile(arr, 0.33) == np.quantile(arr, 0.33)
    assert my_quantile(arr, 0.25) == np.quantile(arr, 0.25)
    assert my_quantile(arr, 0.75) == np.quantile(arr, 0.75)
...