python linspace函数_c语言中的python linspace++

这样怎么样：#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)