将两个矢量<double>转换成矢量<complex <double>>

Question

我有两个类型为double的矢量，我想将它们组合起来形成一个复数矢量。

vector<double> vReal; 
vector<double> vImag; 

如何合并上述两个获得

vector<complex<double>> vComp; 

有人可以帮我我该怎么做呢？

很多谢谢。

问候

Chintan

Answer 1

for(int i = 0; i < vReal.size(); ++i) { 
    complex<double> iNum(vReal[i], vImag[i]); 
    vComp.push_back(iNum); 
} 

还是我失去了一些东西？

也许还有一个检查，以确保vReal和vImag是相同的大小，否则你会崩溃。

Answer 2

使用std::transform，提供合适的BinaryOperation。

Answer 3

vComp.reserve(vReal.size()); 

    std::transform(
     begin(vReal), end(vReal), begin(vImag), 
     std::back_inserter(vComp), 
     [](double r, double i) { return std::complex<double>(r, i); }); 

Answer 4

如果C++ 11

std::complex<double> make_complex(double re, double im) { 
    return std::complex<double>(re,im); 
} 

std::transform(vReal.begin(), vReal.end(), vImag.begin(), std::back_inserter(vComp), make_complex);