Python：手工生成两个环的二维数据

最新推荐文章于 2022-05-12 09:30:53 发布

DeniuHe

最新推荐文章于 2022-05-12 09:30:53 发布

阅读量452

点赞数

分类专栏： Python学习

本文链接：https://blog.csdn.net/DeniuHe/article/details/102931150

版权

Python学习专栏收录该内容

239 篇文章 14 订阅

订阅专栏

import numpy as np
import pandas as pd
import math
import matplotlib.pyplot as plt

num_data = 1000
center = np.array([5.0,5.0])
radiu_inner = 2
radiu_outer = 4

num_inner = num_data //4
num_outer = num_data - num_inner

data = []
theta = 0.0
for i in range(num_inner):
    pho = (theta % 360) * math.pi / 180
    tmp = np.zeros(2,dtype=float)
    tmp[0] = radiu_inner * math.cos(pho) + np.random.rand(1) + center[0]
    tmp[1] = radiu_inner * math.sin(pho) + np.random.rand(1) + center[1]
    data.append(tmp)
    theta += 3
for j in range(num_outer):
    pho = (theta % 360) * math.pi / 180
    tmp = np.zeros(2,dtype=float)
    tmp[0] = radiu_outer * math.cos(pho) + np.random.rand(1) + center[0]
    tmp[1] = radiu_outer * math.sin(pho) + np.random.rand(1) + center[1]
    data.append(tmp)
    theta += 2

label = np.zeros(num_data)
for i in range(num_inner,num_data):
    label[i] = 1
data = np.array(data)

print(data.shape)
print(label.shape)

label = np.vstack(label)
X = np.hstack((data,label))
# X = np.concatenate((data,label.T),axis=0)
print(X.shape)

plt.scatter(X[:,0],X[:,1],c=X[:,2])
plt.show()

X = pd.DataFrame(X)
X.to_csv(r'E:\dataset\clusterData\two_circles.csv',header=None,index=None)