C++之创建等间隔数组

在 C++ 中,数组是一种基本的数据结构,用于存储固定大小的相同类型的元素。数组在声明时必须指定其大小,并且一旦创建,其大小就不可改变。数组的索引从 0 开始。

复制代码
int myArray[5]; // 声明一个整型数组,包含5个元素int anotherArray[5] = {1, 2, 3, 4, 5}; // 声明并初始化一个整型数组

虽然传统的 C++ 数组在声明时大小固定,但 C++ 标准库提供了 std::vector 类,它是一个动态数组,可以在运行时调整大小。

复制代码
#include <vector>std::vector<int> myVector; // 声明一个动态数组myVector.push_back(10); // 添加元素

在 C++ 中,没有内置的函数直接对应于 NumPy 的 linspace 或 arange,但可以通过结合使用标准库中的函数来实现类似的功能。以下是两种方法:

1. 使用 std::vector 和循环来创建等间隔的数组(类似 linspace)

复制代码
#include <iostream>#include <vector>#include <algorithm> // for std::transform​std::vector<double> linspace(double start, double end, int num) {    std::vector<double> linspaced;    if (num == 0) { return linspaced; }    if (num == 1) { linspaced.push_back(start); return linspaced; }    double delta = (end - start) / (num - 1);    linspaced.push_back(start);    std::generate_n(std::back_inserter(linspaced), num - 1, [&]() {        double next = start + delta;        delta = (end - start) / (num - 1);        start = next;        return next;    });    linspaced.push_back(end); // Ensure the end is included    return linspaced;}​int main() {    auto vec = linspace(0, 1, 5);    for (auto v : vec) {        std::cout << v << " ";    }    std::cout << std::endl;    return 0;}

2. 使用 std::vector 和 std::iota 来创建一个序列(类似 arange)

在 C++11 之前,std::iota 是定义在 <algorithm> 头文件中的,但 C++11 之后,它被移到了 <numeric>。

复制代码
#include <iostream>#include <vector>#include <numeric> // for std::iota​std::vector<int> arange(int start, int end, int step) {    std::vector<int> range;    range.reserve((end - start) / step + 1); // 预分配足够的空间    int value = start;    while (value < end) {        range.push_back(value);        value += step;    }    return range;}​int main() {    auto vec = arange(0, 10, 2);    for (auto v : vec) {        std::cout << v << " ";    }    std::cout << std::endl;    return 0;}

补充:

  • **np.linspace(start, stop, num):**返回一个从 start 开始到 stop 结束的等间隔数字的数组,包括结束值,总共 num 个数字。

  • **np.arange(start, stop, step):**返回一个从 start 开始到 stop 结束(不包括 stop),以 step 为步长的等间隔数字的数组。

相关推荐
蹊黎1 小时前
C++中vector类的部分简单实现
开发语言·c++·visual studio
xier_ran8 小时前
【infra之路】AI 编译器学习笔记(六):重拾 C++——lambda 与 MLIR 中的回调
c++·笔记·学习
jimy110 小时前
c++的 Type Traits (一)
开发语言·c++
别动我齐刘海10 小时前
Linux 系统调试学习2
linux·运维·c++·人工智能·学习·机器学习·机器人
ZZHow102412 小时前
Ascend C算子开发(高级)笔记
c++·人工智能·笔记·深度学习·华为云
aqiu11111112 小时前
【C++算法打怪专栏】LeetCode 217. 存在重复元素(解法总结)
c++·算法·leetcode
lojox13 小时前
深入理解 C++ STL:vector 底层实现、迭代器失效与 list 特性
开发语言·c++
微笑星河q13 小时前
C++类和对象(下)初始化列表、类型转换、static、友元、内部类、匿名对象
开发语言·c++
|_⊙14 小时前
C++ 类和对象零散知识点
开发语言·c++
xiangyun6114 小时前
C++ list 模拟实现(二):const迭代器与反向迭代器,模板的黑魔法
开发语言·数据库·c++·算法·stl·list