回答此问题可获得 20 贡献值,回答如果被采纳可获得 50 分。
<p>我在同一轴线上绘制了一系列均方位移(MSD)对数图,用于模拟自推进粒子。模拟运行时,传递标准偏差<code>Dtrans = [0.1, 0.3, 1.0, 3.0, 10]</code>和旋转标准偏差<code>Drot = [0.1, 0.3, 1.0, 3.0, 10, 30, 100, 300]</code>的值不同,总共给出5×8=40个模拟。以下是我目前拥有的:</p>
<pre><code>import numpy as np
import matplotlib.pyplot as plt
import matplotlib.colors as colors
import matplotlib.cm as cmx
import itertools
# List of colour maps
cmaps = ['YlOrRd', 'Greens', 'Blues', 'Purples','RdPu']
nparticles = 5
npts=1000
step=1000
dt = 0.0001
Drot = [0.1, 0.3, 1.0, 3.0, 10, 30, 100, 300]
Dtrans = [0.1, 0.3, 1.0, 3.0, 10]
plt.figure()
plt.title('MSD against time')
plt.xlabel('Time')
plt.ylabel('MSD')
# Time steps along horizontal axis:
timeval = np.linspace(0,dt*step*(npts-1),npts)
# For each value of Dtrans
for d1 in range(len(Dtrans)):
# Choose a colour map:
colourmap = plt.get_cmap(cmaps[d1])
# Limit it to the middle 60%
colourmap = colors.ListedColormap(colourmap(np.linspace(0.2, 0.8, 256)))
# Get a colour from the map for each value of Drot:
values = range(len(Drot))
cNorm = colors.Normalize(vmin=0,vmax=values[-1])
scalarMap=cmx.ScalarMappable(norm=cNorm,cmap=colourmap)
for d2 in range(len(Drot)):
Dr = Drot[d2]
Dt = Dtrans[d1]
MSD = np.zeros((npts,))
x = np.zeros((npts,nparticles))
y = np.zeros((npts,nparticles))
for k in range(npts):
data0=np.loadtxt(open("./Lowdensity/Drot_"+str(Dr)+"/Dtrans_"+str(Dt)+"/ParticleData/ParticleData"+str(k*step)+".csv",'rb'),delimiter=',')
x0,y0= data0[:nparticles,1],data0[:nparticles,2]
x[k,:]=x0
y[k,:]=y0
for k in range(npts):
MSD[k] = np.mean(np.mean((x[k:npts,:] - x[0:(npts-k),:])**2 + (y[k:npts,:] - y[0:(npts-k),:])**2,axis=0))
colorVal = scalarMap.to_rgba(values[d2])
plt.loglog(timeval,MSD,color=colorVal)
plt.legend([('Dtrans='+str(i)+', Drot='+str(j)) for [i,j] in np.array(list(itertools.product(Dtrans,Drot)))])
#plt.loglog(timeval, MSDthry(timeval, 0.5, 100, 70, dt*step*npts))
plt.show()
</code></pre>
<p><a href="https://i.stack.imgur.com/HJzFH.png" rel="nofollow noreferrer"><img src="https://i.stack.imgur.com/HJzFH.png" alt="log-log plot of MSD against time with nasty legend"/></a></p>
<p>彩色地图效果很好,但我现在的传说真的很可怕,不适合这个情节。理想情况下,我希望图例是垂直排列的五个色条,沿着垂直轴有<code>Dtrans</code>个值,水平方向有<code>Drot</code>个值。我如何实现这一点?谢谢</p>