如何使用remove_if與矢量<point2f>

Question

我有一個向量，有很多NaN的x，y位置，我想刪除（做一些opencv工作）。我無法弄清楚如何使用remove_if來刪除NaN（當與擦除一起使用時）。我看過很多例子，如果vector是float或int而不是point2f。任何簡單的例子都會非常有幫助。謝謝。

Answer 1

您可以使用lambda函數，函子或函數指針。這是一個lambda函數的例子：

#include <opencv2/opencv.hpp> 
#include <algorithm> 
#include <iostream> 
#include <cmath> 

using namespace cv; 
using namespace std; 

int main(int argc, char ** argv) 
{ 
    vector<Point2f> pts{ Point2f(1.f, 2.f), Point2f(3.f, sqrt(-1.0f)), Point2f(2.f, 3.f) }; 

    cout << "Before" << endl; 
    for (const auto& p : pts) { 
     cout << p << " "; 
    } 
    cout << endl; 

    pts.erase(remove_if(pts.begin(), pts.end(), [](const Point2f& p) 
    { 
     // Check if a coordinate is NaN 
     return isnan(p.x) || isnan(p.y); 
    }), pts.end()); 

    cout << "After" << endl; 
    for (const auto& p : pts) { 
     cout << p << " "; 
    } 
    cout << endl; 

    return 0; 
} 

，將打印：

Before 
[1, 2] [3, -1.#IND] [2, 3] 
After 
[1, 2] [2, 3]