Hi,
I have an array defined by 3 variables p(x,z,t). I would like to produce
a surface plot with colors defined by p and animate it. That is plot the
value of p at all x and z, over time (t). My code to get p is below but
I really have no idea how to plot this. Anyone know the best way to go
about this?
thanks,
D
# 2D Finite Distance Wave Equation.
from pylab import *
from numpy import math
ion()
# Set up variables.
nx = 100
nz = 100
nsteps = 300
c = 3500
dt = 10**-4
h = 1
t = arange(0,nsteps,dt)
# Define source as a spike.
s = zeros(nsteps)
s[1] = 1
s[2] = 2
s[3] = 1
# Position source.
xs = 50
zs = 50
##plot(t,s)
##show()
# Set up pressure field.
p=empty([nx,nz,nsteps])
for t in range(0,nsteps-1):
for z in range(0,nz-1):
for x in range(0,nx-1):
p[x,z,t] = 0
# Solve wave equation.
for t in range(2,nsteps-1):
for z in range(1,nz-1):
for x in range(2,nx-1):
p[xs,zs,t] = s[t]
k = (c*dt/h)**2
p[x,z,t] = 2*p[x,z,t-1] - p[x,z,t-2] +
k*(p[x+1,z,t-1]-4*p[x,z,t-1]+p[x-1,z,t-1]+p[x,z+1,t-1]+p[x,z-1,t-1])
#Plot somehow
draw()
#close()
------------------------------------------------------------------------------
Virtualization & Cloud Management Using Capacity Planning
Cloud computing makes use of virtualization - but cloud computing
also focuses on allowing computing to be delivered as a service.
http://www.accelacomm.com/jaw/sfnl/114/51521223/
_______________________________________________
Matplotlib-users mailing list
[email protected]
https://lists.sourceforge.net/lists/listinfo/matplotlib-users