本文展示了如何使用C++编写一个模板函数`linspace`，该函数根据给定的起始值、结束值和数量生成等差数列。同时，通过实例比较了C++实现与Numpy库的相同功能的输出结果，强调了输入值与输出数组的精确性。
这样怎么样：#include
#include
template
std::vector linspace(T start_in, T end_in, int num_in)
{
std::vector linspaced;
double start = static_cast(start_in);
double end = static_cast(end_in);
double num = static_cast(num_in);
if (num == 0) { return linspaced; }
if (num == 1)
{
linspaced.push_back(start);
return linspaced;
}
double delta = (end - start) / (num - 1);
for(int i=0; i < num-1; ++i)
{
linspaced.push_back(start + delta * i);
}
linspaced.push_back(end); // I want to ensure that start and end
// are exactly the same as the input
return linspaced;
}
void print_vector(std::vector vec)
{
std::cout << "size: " << vec.size() << std::endl;
for (double d : vec)
std::cout << d << " ";
std::cout << std::endl;
}
int main()
{
std::vector vec_1 = linspace(1, 10, 3);
print_vector(vec_1);
std::vector vec_2 = linspace(6.0, 23.4, 5);
print_vector(vec_2);
std::vector vec_3 = linspace(0.0, 2.0, 1);
print_vector(vec_3);
std::vector vec_4 = linspace(0.0, 2.0, 0);
print_vector(vec_4);
return 0;
}
C++结果：size: 3
1 5.5 10
size: 5
6 10.35 14.7 19.05 23.4
size: 1
0
size: 0
Numpy结果：In [14]: np.linspace(1, 10, 3)
Out[14]: array([ 1. , 5.5, 10. ])
In [15]: np.linspace(6, 23.4, 5)
Out[15]: array([ 6. , 10.35, 14.7 , 19.05, 23.4 ])
In [16]: np.linspace(0.0, 2.0, 1)
Out[16]: array([ 0.])
In [17]: np.linspace(0.0, 2.0, 0)
Out[17]: array([], dtype=float64)
扫一扫
1万+
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
