用matplotlib画三维图的时候我已经设定了z的表达式两个变量，为什么还是报错？

snowJR

import matplotlib.pyplot as plt
from mpl_toolkits.mplot3d import Axes3D
import numpy as np

fig=plt.figure()
ax=Axes3D(fig)
x=np.arange(0,1,1)
y=np.arange(0,8,2)
z=((35*(np.cos(4*x))+20*(np.sin(2*x))+9)/64)*(np.cos(y))
ax.plot_surface(x, y, z)
plt.show()


代码如上，感谢各位大哥

塔利班

1. 
   
2. import matplotlib.pyplot as plt
   
3. from mpl_toolkits.mplot3d import Axes3D
   
4. import numpy as np
   
5. 
   
6. fig=plt.figure()
   
7. ax=Axes3D(fig)
   
8. x=np.arange(0,1,1)
   
9. y=np.arange(0,8,2)
   
10. z=((35*(np.cos(4*x))+20*(np.sin(2*x))+9)/64)*(np.cos(y)).reshape(-1,1)
    
11. print(z.shape)
    
12. ax.plot_surface(x, y, z)
    
13. plt.show()

复制代码

snowJR

塔利班 发表于 2020-2-18 11:53

这样出来的图像不是我要结果了

塔利班

snowJR 发表于 2020-2-18 11:55
这样出来的图像不是我要结果了

你想要打的是什么

snowJR

塔利班 发表于 2020-2-18 12:01
你想要打的是什么

以x，y为变量z为函数的图像，我插一个用sympy画的给你看看
import sympy as sp
from sympy.plotting import plot3d
x=sp.symbols('x')
y=sp.symbols('y')
x1=sp.sin(x)
x2=sp.cos(4*x)
x3=sp.cos(2*x)
y1=sp.cos(y)
plot3d(((35*x2+20*x3+9)/64)*y1,(x,0,4),(y,0,8))

塔利班

1. 
   
2. import matplotlib.pyplot as plt
   
3. from mpl_toolkits.mplot3d import Axes3D
   
4. import numpy as np
   
5. 
   
6. fig=plt.figure()
   
7. ax=Axes3D(fig)
   
8. x=np.arange(0,10,0.1)
   
9. y=np.arange(0,8,0.1)
   
10. x,y = np.meshgrid(x,y)
    
11. z=((35*(np.cos(4*x))+20*(np.sin(2*x))+9)/64)*(np.cos(y))
    
12. ax.plot_surface(x, y, z)
    
13. plt.show()

复制代码

snowJR

塔利班 发表于 2020-2-18 12:18

x,y = np.meshgrid(x,y)在这里起到什么作用?

塔利班

snowJR 发表于 2020-2-18 12:39
x,y = np.meshgrid(x,y)在这里起到什么作用?

之前的x,y没有任何关系，这里是建立网格点，一个xoy上的点得有x,y坐标吧，要不你x,y长度也不一样

snowJR

塔利班 发表于 2020-2-18 12:55
之前的x,y没有任何关系，这里是建立网格点，一个xoy上的点得有x,y坐标吧，要不你x,y长度也不一样

哦哦，明白了，谢谢大佬~

snowJR

塔利班 发表于 2020-2-18 12:55
之前的x,y没有任何关系，这里是建立网格点，一个xoy上的点得有x,y坐标吧，要不你x,y长度也不一样

大佬大佬，我改了一下代码，画的图超出坐标轴了。。。。
import matplotlib.pyplot as plt
from mpl_toolkits.mplot3d import Axes3D
import numpy as np

fig=plt.figure()
ax=Axes3D(fig)
x=np.arange(0,10,0.1)
y=np.arange(0,8,0.1)
x,y = np.meshgrid(x,y)
z=((35*(np.cos(4*x))+20*(np.sin(2*x))+9)/64)*(np.cos(y))

ax.set_xlabel('θ')
ax.set_xlim(0,4)
ax.set_ylabel('φ')
ax.set_ylim(0,8)
ax.set_zlabel('Y')
ax.set_zlim(0,1)
ax.plot_surface(x, y, z)
plt.show()