[已解决]求助

snowJR

老铁们，写了一个画图的程序，老是提示我 'Y' is not defined，能帮我看看是哪一步做错了吗？
#引入第三方库
import matplotlib.pyplot as plt
from mpl_toolkits.mplot3d import Axes3D
import numpy as np

#建立三维数据
θ=np.arange(0,np.pi,0.1)
φ=np.arange(0,2*np.pi,0.1)
θ,φ = np.meshgrid(θ,φ)

#球函数字典
dic1={
        0:np.sin(φ),
        1:(np.cos(θ))*(np.sin(φ)),
        2:((3*(np.cos(2*θ))+1)/4)*(np.sin(φ))
        }
dic2={
        1:(np.cos(θ))*(np.sin(φ)),
        2:3*(np.sin(θ))*(np.cos(θ))*(np.sin(φ)),
        4:3*(np.sin(θ))*(np.sin(θ))*(np.sin(φ))
        }

#定义三维坐标轴
fig=plt.figure()
ax=Axes3D(fig)

def main():
        l=input('请输入阶数l:')
        m=input('请输入m:')
        x = int(l)
        y = int(m)
        t=x*y
        Y= sp_main(t)


#球函数的阶数
def sp_main(t):
        if t==0:
                Y=dic1.get(t)
                Y= np.maximum(Y, 0)
               
        else:
                Y=dic2.get(t)
                Y= np.maximum(Y, 0)
        return Y       




#作图
plt.rcParams['font.sans-serif']=['SimHei']
ax.set_xlabel('仰角θ')
ax.set_ylabel('方位角φ')
ax.set_zlabel('球函数Y')

ax.set_xlim(0,np.pi)
ax.set_ylim(0,2*np.pi)
ax.set_zlim(0,1)
ax.plot_surface(θ, φ, Y)

plt.show()

if __name__ == "__main__":
    main()

最佳答案

月排行榜 / 总排行榜

永恒的蓝色梦想

2020-8-8 18:27:45

snowJR 发表于 2020-8-8 18:19
这个为什么没有图像出来？

你写的代码，问我咯？

跳转到最佳答案楼层

sunrise085

ax.plot_surface(θ, φ, Y)
这一行的Y的确没有被定义过

在球函数的阶数中出现过 Y，但是那是局部变量。
此外就没有见过Y的定义了

永恒的蓝色梦想

1. #引入第三方库
   
2. import matplotlib.pyplot as plt
   
3. from mpl_toolkits.mplot3d import Axes3D
   
4. import numpy as np
   
5. 
   
6. #建立三维数据
   
7. θ=np.arange(0,np.pi,0.1)
   
8. φ=np.arange(0,2*np.pi,0.1)
   
9. θ,φ = np.meshgrid(θ,φ)
   
10. 
    
11. 
    
12. #球函数字典
    
13. dic1={
    
14.     0:np.sin(φ),
    
15.     1:(np.cos(θ))*(np.sin(φ)),
    
16.     2:((3*(np.cos(2*θ))+1)/4)*(np.sin(φ))
    
17. }
    
18. 
    
19. 
    
20. dic2={
    
21.     1:(np.cos(θ))*(np.sin(φ)),
    
22.     2:3*(np.sin(θ))*(np.cos(θ))*(np.sin(φ)),
    
23.     4:3*(np.sin(θ))*(np.sin(θ))*(np.sin(φ))
    
24. }
    
25. 
    
26. 
    
27. #定义三维坐标轴
    
28. fig=plt.figure()
    
29. ax=Axes3D(fig)
    
30. 
    
31. 
    
32. #球函数的阶数
    
33. def sp_main(t):
    
34.     if t==0:
    
35.         Y=dic1.get(t)
    
36.         Y= np.maximum(Y, 0)
    
37.            
    
38.     else:
    
39.         Y=dic2.get(t)
    
40.         Y= np.maximum(Y, 0)
    
41.     return Y      
    
42. 
    
43. 
    
44. 
    
45. def main():
    
46.     l=input('请输入阶数l:')
    
47.     m=input('请输入m:')
    
48.     x = int(l)
    
49.     y = int(m)
    
50.     t=x*y
    
51.     Y= sp_main(t)
    
52.     #作图
    
53.     plt.rcParams['font.sans-serif']=['SimHei']
    
54.     ax.set_xlabel('仰角θ')
    
55.     ax.set_ylabel('方位角φ')
    
56.     ax.set_zlabel('球函数Y')
    
57. 
    
58.     ax.set_xlim(0,np.pi)
    
59.     ax.set_ylim(0,2*np.pi)
    
60.     ax.set_zlim(0,1)
    
61.     ax.plot_surface(θ, φ, Y)
    
62. 
    
63.     plt.show()
    
64. 
    
65. if __name__ == "__main__":
    
66.     main()

复制代码

snowJR

永恒的蓝色梦想 发表于 2020-8-8 18:00

这个为什么没有图像出来？

永恒的蓝色梦想

snowJR 发表于 2020-8-8 18:19
这个为什么没有图像出来？

你写的代码，问我咯？