Вычислить скользящее среднее с помощью ojalgo - PullRequest
0 голосов
/ 26 апреля 2018

Каков наилучший способ вычисления скользящего среднего с использованием ojalgo?

Первый SampleSet добавляет каждое число столько раз, сколько размер окна.

Во-вторых, цель кода не очевидна.

У меня есть:

    final Array1D<Double> doubles = Array1D.factory(Primitive32Array.FACTORY).makeFilled(10, new Uniform());
    System.out.println(doubles.toString());

    final Array1D<Double> rollingMedian = Array1D.PRIMITIVE32.makeZero(doubles.size());

    rollingMedian.set(0, doubles.get(0));
    System.out.printf("%1$s -> %1$s\n", doubles.get(0));

    final Array1D<Double> someSamples2 = doubles.subList(0, 2);
    final double mean2 = SampleSet.wrap(someSamples2).getMean();
    rollingMedian.set(1, mean2);
    System.out.printf("%s -> %s\n", someSamples2.toString(), mean2);

    for (int i = 2; i < doubles.length; i++) {
        final Array1D<Double> someSamples = doubles.subList(i - 2, i + 1);
        final double mean = SampleSet.wrap(someSamples).getMean();
        rollingMedian.set(i, mean);
        System.out.printf("%s -> %s\n", someSamples.toString(), mean2);
    }

    System.out.println(rollingMedian.toString());

Что не самое приятное ...

1 Ответ

0 голосов
/ 26 апреля 2018

Не уверен, что будет самым хорошим способом, но вот один из вариантов:

    for (int i = 0; i < doubles.length; i++) {
        int first = Math.max(0, i - 2);
        int limit = i + 1;
        double mean = doubles.aggregateRange(first, limit, Aggregator.AVERAGE);
        rollingMedian.set(i, mean);
    }
...