基于python实现的双月模型

刚好用到了双月模型，百度了一下发现并没有很好的双月模型，就自己尝试写了一下，欢迎大家批评指正

# -*- 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()