使用python 的matplotlib 画轨道实例

yipeiwu_com6年前Python基础

如下所示:

import numpy as np 
import matplotlib.pyplot as plt
import matplotlib.patches as mpatches
from scipy import stats
 
fig = plt.figure()
ax = fig.add_subplot(111, xlim=(0, 10), ylim=(-4, 4))
sx=0;sy=0;r=1.5 ; 
circle = mpatches.Circle((sx,sy),r,ec='b',fc = 'b',alpha=0.6)
ax.add_patch(circle)
n = 5
alpha = np.linspace(-np.pi/2,np.pi/2,100)
for i in range(1,n+1):
 a = (i+1)*2
 b = (i+1)
 plt.plot(a*np.cos(alpha),b*np.sin(alpha))
 if i <= 3:
  ax.text(a+1,0,'layer.%d'%(i-1),ha='center',va='center')
rate = 3
Jnum = 100
for i in range(Jnum):
 data = stats.poisson.rvs(mu = rate, loc = 0 , size = 1)
 while data == 0:
  data = stats.poisson.rvs(mu = rate, loc = 0 , size = 1)
 data = data[0] 
 r = np.random.random()
 beta = (np.random.random()-0.5)*np.pi 
 print beta
 data += r
 a = (data+1)*2*np.cos(beta)
 b = (data+1)*np.sin(beta)
 ax.scatter(a,b, c='y',edgecolors = 'y')
xx = [3.5 , 6.8,6.8]
yy = [-1.8,0.9,3.1]
pp = [1.00,0.85,0.75]
rota = [-45,0,0]
for i in range(3):
 plt.plot((1.5,xx[i]) , (0,yy[i]) , 'r')
 ax.scatter(xx[i],yy[i], c='y',edgecolors = 'b')
 ax.text(xx[i] , yy[i] , 'probability %lf'%pp[i] , ha = 'center' )
tx = np.linspace(0,10,10)
ty = np.linspace(-4,-4,10)
ax.set_xticks(tx)
ax.set_xticklabels((''))
ax.set_yticks(ty)
ax.set_yticklabels((''))
#plt.axis('off')
plt.show()

以上这篇使用python 的matplotlib 画轨道实例就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持【听图阁-专注于Python设计】。

相关文章

用Python写一段用户登录的程序代码

用Python写一段用户登录的程序代码

如下所示: #!/usr/bin/env python #coding: utf8 import getpass db = {} def newUser(): username =...

详解Python中的内建函数,可迭代对象,迭代器

详解Python中的内建函数,可迭代对象,迭代器

Python中的内建函数和可迭代对象,迭代器 求值标识 id() #标识id 返回对象的唯一标识,CPython返回内存地址 hash() #哈希, 返回对象的哈希值 le...

python各类经纬度转换的实例代码

python各类经纬度转换的实例代码

python各类经纬度转换,具体代码如下所示: import math import urllib import json x_pi = 3.14159265358979324 *...

Django模板导入母版继承和自定义返回Html片段过程解析

1.ROOT_URLCONF = '总路由所在路径(比如untitled.urls)'<===默认情况是这样 根路由的路径是可以修改的:ROOT_URLCONF = app01.u...

使用PyTorch实现MNIST手写体识别代码

实验环境 win10 + anaconda + jupyter notebook Pytorch1.1.0 Python3.7 gpu环境(可选) MNIST数据集介绍 MNIST 包...