刚好用到了双月模型,百度了一下发现并没有很好的双月模型,就自己尝试写了一下,欢迎大家批评指正
# -*- coding: utf-8 -*-
"""
Created on Fri Jan 25 15:06:07 2019
@author: WuYan
"""
import numpy as np
import matplotlib.pyplot as plt
x = np.random.random([100, 1])
def moon(N = 10000, r = 10, w = 3): #width为双月模型的宽度一半
data_alpha = np.random.random([N,1]) * np.pi
s = np.sign(np.random.random([N,1]) - 0.5)
width = s * w * np.random.random([N,1])
pho = r + width
data_x, data_y = pho * np.cos(data_alpha), pho * np.sin(data_alpha)
data = np.concatenate((data_x, data_y), axis=1)
return data
def double_moon(N = 10000, r = 10, w = 3, d = 2): #此处d可以设为负数,表述双月有重叠区域
moon_1 = moon(N, r, w)
moon_2 = -moon_1
moon_2[:,0] = moon_2[:,0] + r
moon_2[:,1] = moon_2[:,1] - d
return moon_1,moon_2
a, b= double_moon()
plt.plot(a[:,0],a[:,1], 'r*', b[:,0],b[:,1], 'b*')
plt.show()