将较小长度的向量返回到已填充的向量

Question

我有一个填充了 x 个值的向量，但我需要使用初始向量多次迭代函数，但每次返回一个可能更少值的向量。这是我目前要做的，目前向量保留所有值，我不确定如何覆盖初始向量声明或有效地从该初始向量中删除项目。

for (auto& image : images)
    {
        rValues.push_back(image->pixels[pix].r);
        gValues.push_back(image->pixels[pix].g);
        bValues.push_back(image->pixels[pix].b);
    }
    for (int i = 0; i < 5; i++)
    {
        float rMedian = calculateMedian(rValues);
        float rsD = calculateStandardDeviation(rValues);
        rValues = sigmaClip(rValues, rsD, rMedian);


        for (auto& r : rValues)
        {
            std::cout << "r val : " << r << "\n";
        }
        system("pause");
    }

（忽略 rValues 的 std::cout，使用它来检查每次迭代是否删除了值）。

这就是我在 sigmaClip 函数中实现的，它需要返回不同长度的向量。

std::vector<float> sigmaClip(std::vector<float> values, float sD, float median)
{
    //Sigma Clip
    float upperBound = median + sD;
    float lowerBound = median - sD;
    std::vector<float> returnVals;
    for (auto& val : values)
    {
        if ((val > lowerBound) && (val < upperBound))
            returnVals.push_back(val);
    }

    return returnVals;
}

我最初尝试在传递给sigmaClip() 的向量上使用 .erase() 和 remove_if 但我认为这些函数没有按我的预期工作，因此在sigmaClip() 中创建另一个向量并返回它希望它会覆盖初始 rValues 向量。

Answer 1

std::remove_if 如果与erase 方法配合使用以实现the erase-remove idiom，则应该可以很好地完成工作。重新定义 sigmaClip 以接受对 values 的可变引用（而不是返回它），例如：

void sigmaClip(std::vector<float>& values, float sD, float median)
{
    //Sigma Clip
    const float upperBound = median + sD;
    const float lowerBound = median - sD;
    values.erase(std::remove_if(std::begin(values), std::end(values),
                                [=](float val) { return val < lowerBound || val > upperBound; }),
                 std::end(values));
}

应该可以正常工作，将调用站点更改为：

// Remove assignment back to rValues
sigmaClip(rValues, rsD, rMedian);