Python matplotlib三维线框图
好的,我有一个x值,y值和z值的列表(我想我把它们转换成了数组?),我想做一个曲面图,但它不起作用 这就是我要做的,你可以忽略大部分代码,因为它是非常不相关的-只要看看结尾,我有Python matplotlib三维线框图,python,3d,numpy,matplotlib,Python,3d,Numpy,Matplotlib,好的,我有一个x值,y值和z值的列表(我想我把它们转换成了数组?),我想做一个曲面图,但它不起作用 这就是我要做的,你可以忽略大部分代码,因为它是非常不相关的-只要看看结尾,我有xdis,ydis和dist,在我试图绘制atm时,我得到了value错误:需要超过1个值才能解包:(.非常感谢帮助 from math import * from numpy import * import pylab def sweep (v,p,q,r,s): a=.98 for i in ran
xdis
,ydis
和dist
,在我试图绘制atm时,我得到了value错误:需要超过1个值才能解包:(.非常感谢帮助
from math import *
from numpy import *
import pylab
def sweep (v,p,q,r,s):
a=.98
for i in range (1, len(v)-1):
for j in range (1, len(v)-1):
c =0.0
if i==p and j==q: c =1.0
if i==r and j==s: c= -1.0
v[i,j]=(v[i -1,j]+v[i+1,j]+v[i,j -1]+v[i,j+1]+c-a*v[i,j])/(4-a)
def main():
from mpl_toolkits.mplot3d import Axes3D
from matplotlib import cm
from matplotlib.ticker import LinearLocator, FormatStrFormatter
import matplotlib.pyplot as plt
ydis=[]
xdis=[]
resis=[]
for j in range(2,18):
for i in range(2,18):
v= zeros ((20,20),float )
p=q=9
r=i
s=j
dv =1.0e10
lastdv =0
count =0
while (fabs(dv - lastdv)>1.0e-7*fabs(dv)):
lastdv =dv
sweep(v,p,q,r,s)
dv=v[p,q]-v[r,s]
resis.append(dv)
xdis.append(r-p)
ydis.append(s-q)
X=array(xdis)
Y=array(ydis)
Z=array(resis)
fig = plt.figure()
ax = fig.add_subplot(111, projection='3d')
ax.plot_wireframe(X,Y,Z)
plt.show()
main()
“sweep”函数似乎没有修改“v”,因此您得到的是一个空列表。需要三个2D数组(X、Y、Z)作为输入。因此
之后:
X=np.array(xdis)
Y=np.array(ydis)
Z=np.array(resis)
加:
X=X.reshape((-1,16))
Y=Y.reshape((-1,16))
Z=Z.reshape((-1,16))