將兩個矢量<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);