无标题文章

from mpl_toolkits.mplot3d import Axes3D

from matplotlib import cm

from matplotlib.ticker import LinearLocator, FormatStrFormatter

import matplotlib.pyplot as plt

import numpy as np

import random

N=100

accuracy=1e-10

class Fields:

def init(self):

self.v=[]

self.old_v=[]

for i in range(N):

self.v.append([])

self.old_v.append([])

print len(self.old_v)

for i in range(N):

for j in range(N):

self.v[i].append(0)

self.old_v[i].append(0)

for i in range(N/4,3*N/4):

for j in range(N/4,3*N/4):

self.v[i][j]=1

def update(self):

self.Delta_v=0.

for i in range(N):

self.v[i][0]=0

self.v[i][N-1]=0

for j in range(N):

self.v[0][j]=0

self.v[N-1][j]=0

for i in range(N):

for j in range(N):

self.old_v[i][j]=self.v[i][j]

for i in range(1,N-1):

for j in range(1,N-1):

self.v[i][j]=(self.old_v[i+1][j]+self.old_v[i-1][j]+self.old_v[i][j+1]+self.old_v[i][j-1])/4.

print self.Delta_v

def fire(self):

counter=0

for i in range(10):

self.update()

counter+=1

i+=1

return self.v

Super=Fields()

Soup=Super.fire()

fig = plt.figure()

ax = fig.gca(projection='3d')

x=np.linspace(-1,1,N)

y=np.linspace(-1,1,N)

x, y = np.meshgrid(x, y)

surf = ax.plot_surface(x, y, Soup, rstride=1, cstride=1, cmap=cm.coolwarm,

linewidth=0, antialiased=False)

ax.set_zlim(-1.01, 1.01)

ax.zaxis.set_major_locator(LinearLocator(10))

ax.zaxis.set_major_formatter(FormatStrFormatter('%.02f'))

plt.xlabel("X")

plt.ylabel("Y")

fig.colorbar(surf, shrink=0.5, aspect=5)

plt.show()

Contact GitHub
API
Training
Shop
Blog
About

[

](https://github.com/)