快速生成一组环形数据

liusuihuashang

已于 2023-11-23 21:28:47 修改

阅读量399

点赞数 8

文章标签：数据库

于 2023-11-23 21:07:22 首次发布

本文链接：https://blog.csdn.net/liusuihuashang/article/details/134586516

版权

本文介绍了如何使用C++和高中数学知识自定义函数生成环形数据，包括控制数据的随机性和误差范围，最后数据以CSV格式存储，以便后续分析。这与sklearn库提供的功能形成对比，展示了编程实现机器学习数据预处理的可能性。

摘要由CSDN通过智能技术生成

sklearn是一个开源的机器学习库，支持不同种类的机器学习算法，并且提供了许多质量良好的数据集。假如我们想要得到一组环形数据集，借助sklearn的包很轻易就可以实现，不过换个角度思考，我们自己动手是否也可以生成一组数据，使之在散点图上环状分布；借助C++的random头文件以及一点高中数学知识，我们很快也可以打造属于自己的数据集。

简单回顾一下，ρ与x和y之间的关系，y=ρsin(θ)，x=ρcos(θ)。这是第一象限的情况，对于其它象限，只需要注意角度和符号的关系便可。

#include <iostream>
#include <fstream>
#include <random>
#include <string>
#include <utility>
#include <cmath>

std::random_device rd;
std::mt19937 g(rd());
std::uniform_real_distribution dis(0.0, M_PI * 2);

std::pair<double, double> circleData(double radius) {
    double tolerance = radius / (30.0 + dis(g));
    int n;
    double d = dis(g);
    if (d < M_PI)
        n = 1;
    else 
        n = -1;

    double r = radius + n * tolerance * dis(g);
    double x, y;
    double sita = dis(g);
    double mapping = sita / (M_PI / 2);

    if (mapping < 1.0) {
        x = r * cos(sita);
        y = r * sin(sita);
    } else if (mapping < 2.0) {
        x = -r * cos(M_PI - sita);
        y = r * sin(M_PI - sita);
    } else if (mapping < 3.0) {
        x = -r * cos(sita - M_PI);
        y = -r * sin(sita - M_PI);
    } else {
        x = r * cos(2 * M_PI - sita);
        y = -r * sin(2 * M_PI - sita);
    }

    return std::pair<double, double>{x, y};
}

void generateData(const std::string& path, std::size_t n = 1000, double radius = 1000.0) {
    std::ofstream out{path};

    out << "X,Y\n";
    for (std::size_t i{}; i < n; ++i) {
        std::pair<double, double> pii = circleData(radius);
        std::string str = std::to_string(pii.first) + ',' + std::to_string(pii.second) + '\n';
        out << str;
    }
}

int main() {
    std::string str{};
    std::cin >> str;
    generateData(str);
}

需要额外补充几点：1.生成的数据并不必完全呈环状，有稍微的偏差更加符合随机性，所以这里定义了tolerance变量，允许在半径范围内有一定的误差。2.生成的数据会写入csv格式的文件当中，而csv格式下的数据说白了就是一堆以逗号作为分割界限的字符串，后面用借助Python的pandas库便能很容易地对csv格式文件进行解析。3.为了确定随机生成的角度属于哪一象限，只需要除以(pi/2)即可判断，浮点数比较带来的精度丢失可接受。